Update app.py

app.py

@@ -1,20 +1,13 @@
 import gradio
 from transformers import ViltProcessor, ViltForQuestionAnswering
 from PIL import Image
-import requests
 
 processor = ViltProcessor.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
 model = ViltForQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
 
 
 def predict_answer(image, question):
-    if isinstance(image, str):
-        # Download image from URL
-        response = requests.get(image)
-        image = Image.open(BytesIO(response.content))
-    else:
-        # Convert numpy array to PIL Image
-        image = Image.fromarray(image.astype('uint8'), 'RGB')
+    image = Image.fromarray(image.astype('uint8'), 'RGB')
     encoding = processor(image, question, return_tensors="pt")
     outputs = model(**encoding)
     logits = outputs.logits
@@ -31,8 +24,7 @@ def predict_answer(image, question):
 
 
 inputs = [
-    gradio.inputs.Image(type="file", label="Upload Image"),
-    gradio.inputs.Image(type="url", label="Image URL"),
+    gradio.components.Image(label="Image"),
     gradio.components.Textbox(label="Question", placeholder="Enter your question here.")
 ]