Update app.py

app.py

@@ -60,9 +60,9 @@ def getProductDetails(history, image):
     # for response in stream:
     #     output += response.token.text
     #     yield [(prompt, output)]
-    # gr.Info('Gemma:' + output)
+    gr.Info('Gemma:' + product_description)
     # history.append((prompt, output))
-    yield history
+    return history
 
 @torch.no_grad()
 def getImageDescription(image):
@@ -120,6 +120,7 @@ def getImageDescription(image):
         # history[-1][1] = partial_response
         # yield history
     gr.Info('Got:' + partial_response)
+    return partial_response
 
 def resize_image(image):
     image = image.convert('RGB')
@@ -178,7 +179,7 @@ with gr.Blocks(css=css) as demo:
             output = gr.Image(type="pil", interactive=False)
         
     response_handler = (
-        getImageDescription,
+        getProductDetails,
         [image],
         []
     )