Update app.py

app.py

@@ -61,8 +61,8 @@ def getProductDetails(history, image):
         do_sample=True,
         seed=rand_val,
     )
-    system_prompt="you're a helpful e-commerce marketting assitant"
-    prompt="Write me a poem"
+    system_prompt="You're a helpful e-commerce marketing assitant."
+    prompt="Our product description is as follows: " + product_description + ". Please write four product title options for this art product."
     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 = ""
@@ -76,15 +76,11 @@ def getProductDetails(history, image):
 
 @torch.no_grad()
 def getImageDescription(image):
-    message = "Generate a product title for the image"
+    message = "Generate an ecommerce product description 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}"