ndtran commited on
Commit
a281098
·
1 Parent(s): 144528c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -95,6 +95,10 @@ model = model.to(device)
95
 
96
  def get_answer(image, url, question) -> str:
97
  global model, device
 
 
 
 
98
 
99
  result = model.inference(image, question, device)
100
  return result.get('answer', 'I don\'t know :<')
@@ -120,7 +124,6 @@ with gr.Blocks() as demo:
120
  ask = gr.Button(label="Get the answer")
121
 
122
  with gr.Column():
123
-
124
  answer = gr.Label(label="Answer")
125
 
126
  ask.click(get_answer, inputs=[image, image_url, question], outputs=[answer])
 
95
 
96
  def get_answer(image, url, question) -> str:
97
  global model, device
98
+
99
+ if url is not None and (url.startswith('http') or url.startswith('https')):
100
+ result = model.inference(load_image_from_URL(url), question, device)
101
+ return result.get('answer', 'I don\'t know :<')
102
 
103
  result = model.inference(image, question, device)
104
  return result.get('answer', 'I don\'t know :<')
 
124
  ask = gr.Button(label="Get the answer")
125
 
126
  with gr.Column():
 
127
  answer = gr.Label(label="Answer")
128
 
129
  ask.click(get_answer, inputs=[image, image_url, question], outputs=[answer])