Spaces:

prateekbh
/

product-description-maker

Runtime error

prateekbh commited on Feb 29

Commit

5da11b3

•

1 Parent(s): eaebf0f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -82,7 +82,6 @@ def interactWithModel(history, prompt):
 @torch.no_grad()
 def getImageDescription(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."}]
@@ -130,7 +129,6 @@ def getImageDescription(image):
         partial_response += new_token
         # history[-1][1] = partial_response
         # yield history
-    gr.Info('Got:' + partial_response)
     return partial_response
 def resize_image(image):

 @torch.no_grad()
 def getImageDescription(image):
     message = "Generate an ecommerce product description for the image"
     stop = StopOnTokens()
     messages = [{"role": "system", "content": "You are a helpful assistant."}]
         partial_response += new_token
         # history[-1][1] = partial_response
         # yield history
     return partial_response
 def resize_image(image):