Update app.py

app.py

@@ -51,8 +51,7 @@ def getProductDetails(history, image):
     product_description=getImageDescription(image)
     system_prompt="You're a helpful e-commerce marketing assitant working on art products."
     prompt="Our product description is as follows: " + product_description + ". Please write a product title options for it."
-    history = interactWithModel(history, system_prompt, prompt)
-    yield history
+    interactWithModel(history, system_prompt, prompt)
 
 def interactWithModel(history, system_prompt, prompt):
     client = InferenceClient("google/gemma-7b-it")
@@ -70,13 +69,10 @@ def interactWithModel(history, system_prompt, prompt):
     formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
-    
     for response in stream:
         output += response.token.text
         yield [(prompt, output)]
-    gr.Info('Gemma:' + product_description)
     history.append((prompt, output))
-    return history
 
 @torch.no_grad()
 def getImageDescription(image):