Update app.py

app.py

@@ -38,18 +38,11 @@ class StopOnTokens(StoppingCriteria):
         return False
 
 @torch.no_grad()
-def response(history, image):
+def response(image):
     message = "Generate a product title for the image"
     gr.Info('Starting...' + message)
     stop = StopOnTokens()
     messages = [{"role": "system", "content": "You are a helpful assistant."}]
-
-    for user_msg, assistant_msg in history:
-        messages.append({"role": "user", "content": user_msg})
-        messages.append({"role": "assistant", "content": assistant_msg})
-
-    if len(messages) == 1:
-        message = f" <image>{message}"
     
     messages.append({"role": "user", "content": message})
 
@@ -85,15 +78,11 @@ def response(history, image):
         )
     t = Thread(target=model.generate, kwargs=generate_kwargs)
     t.start()
-
-    history.append([message, ""])
     partial_response = ""
     for new_token in streamer:
         partial_response += new_token
-        history[-1][1] = partial_response
-        yield history
     gr.Info('Got:' + partial_response)
-    return history
+    return partial_response
 
 def resize_image(image):
     image = image.convert('RGB')
@@ -153,8 +142,8 @@ with gr.Blocks(css=css) as demo:
         
     response_handler = (
         response,
-        [chat, image],
-        [chat]
+        [image],
+        []
     )
 
     background_remover_handler = (