prateekbh commited on
Commit
0f4c15d
1 Parent(s): f2717c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -39,9 +39,9 @@ class StopOnTokens(StoppingCriteria):
39
 
40
  @torch.no_grad()
41
  def response(history, image):
 
42
  gr.Info('Starting...' + message)
43
  stop = StopOnTokens()
44
- message = "Generate a product title for the image"
45
  messages = [{"role": "system", "content": "You are a helpful assistant."}]
46
 
47
  for user_msg, assistant_msg in history:
@@ -145,7 +145,6 @@ with gr.Blocks(css=css) as demo:
145
  with gr.Row():
146
  with gr.Column(elem_id="col-container"):
147
  image = gr.Image(type="pil")
148
- message = gr.Textbox(interactive=True, show_label=False, container=False)
149
  chat = gr.Chatbot(show_label=False)
150
  submit = gr.Button(value="Upload", variant="primary")
151
  with gr.Column():
 
39
 
40
  @torch.no_grad()
41
  def response(history, image):
42
+ message = "Generate a product title for the image"
43
  gr.Info('Starting...' + message)
44
  stop = StopOnTokens()
 
45
  messages = [{"role": "system", "content": "You are a helpful assistant."}]
46
 
47
  for user_msg, assistant_msg in history:
 
145
  with gr.Row():
146
  with gr.Column(elem_id="col-container"):
147
  image = gr.Image(type="pil")
 
148
  chat = gr.Chatbot(show_label=False)
149
  submit = gr.Button(value="Upload", variant="primary")
150
  with gr.Column():