Update app.py

app.py

@@ -93,7 +93,7 @@ model = OCRVQAModel({
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = model.to(device)
 
-def get_answer(image, question) -> str:
+def get_answer(image, url, question) -> str:
     global model, device
     
     result = model.inference(image, question, device)
@@ -112,27 +112,17 @@ with gr.Blocks() as demo:
                 """
             )
             
-            image_url = gr.Textbox(lines=1, label="Image URL", placeholder="Paste image URL here")
-            
-            if image_url.value:
-                print("Loading image from URL...")
-                image = load_image_from_URL(image_url)
-            else:
-                # Or upload from your computer
-                print("Loading uploaded image...")
-                image = gr.Image(shape=(224, 224), type="pil")
-                # image = transforms.ToTensor()(image)
-                # image = transforms.ToPILImage()(image)
-            
-            print(type(image))
+            image = gr.Image(shape=(224, 224), type="pil")
+            image_url = gr.Textbox(lines=1, label="Image URL", placeholder="Or, paste the image URL here")
+            question = gr.Textbox(lines=5, label="Question")
 
-            # print(image.size)
+        with gr.Column():
+            ask = gr.Button(label="Get the answer")
 
         with gr.Column():
-            question = gr.Textbox(lines=5, label="Question")
+            
             answer = gr.Label(label="Answer")
-            ask = gr.Button(label="Get the answer")
 
-    ask.click(get_answer, inputs=[image, question], outputs=[answer])
+    ask.click(get_answer, inputs=[image, image_url, question], outputs=[answer])
 
-    demo.launch()
+    demo.launch(shared = True)