Update app.py

app.py

@@ -65,15 +65,15 @@ def getProductDetails(history, image):
     yield history
 
 @torch.no_grad()
-def getImageDescription(history, image):
+def getImageDescription(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})
+    # 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}"
@@ -113,12 +113,12 @@ def getImageDescription(history, image):
     t = Thread(target=model.generate, kwargs=generate_kwargs)
     t.start()
 
-    history.append([message, ""])
+    # history.append([message, ""])
     partial_response = ""
     for new_token in streamer:
         partial_response += new_token
-        history[-1][1] = partial_response
-        yield history
+        # history[-1][1] = partial_response
+        # yield history
     gr.Info('Got:' + partial_response)
 
 def resize_image(image):
@@ -179,8 +179,8 @@ with gr.Blocks(css=css) as demo:
         
     response_handler = (
         getImageDescription,
-        [chat, image],
-        [chat]
+        [image],
+        []
     )
 
     background_remover_handler = (