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