Update app.py
Browse files
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():
|