prateekbh commited on
Commit
cd87db1
·
verified ·
1 Parent(s): 35d8e54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ from transformers import StoppingCriteria, TextIteratorStreamer, StoppingCriteri
8
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
 
10
 
11
- title = """<h1 style="text-align: center;">Product description generator</h1>"""
12
  css = """
13
  div#col-container {
14
  margin: 0 auto;
@@ -86,7 +86,7 @@ def response(message, history, image):
86
  with gr.Blocks(css=css) as demo:
87
  with gr.Column(elem_id="col-container"):
88
  gr.HTML(title)
89
- image = gr.Image(type="pil", id="upload")
90
  message = gr.Textbox(interactive=True, show_label=False, container=False)
91
  chat = gr.Chatbot(show_label=False)
92
  submit = gr.Button(value="Upload", variant="primary")
 
8
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
 
10
 
11
+ title = """<h1 style="text-align: center;">Product description generator</h1><script src='/assets/upload.js'/>"""
12
  css = """
13
  div#col-container {
14
  margin: 0 auto;
 
86
  with gr.Blocks(css=css) as demo:
87
  with gr.Column(elem_id="col-container"):
88
  gr.HTML(title)
89
+ image = gr.Image(type="pil", elem_id="upload")
90
  message = gr.Textbox(interactive=True, show_label=False, container=False)
91
  chat = gr.Chatbot(show_label=False)
92
  submit = gr.Button(value="Upload", variant="primary")