Update app.py

app.py

@@ -2,24 +2,22 @@ import streamlit as st
 from PIL import Image
 import requests
 from io import BytesIO
-
-# Function to perform image question answering
-def perform_image_qa(image, question):
-    inputs = {"image": image, "question": question}
-    response = hf_spaces.use_model(model_name, inputs)
-    return response['answer']
+from transformers import pipeline
 
 def main():
     st.title("Image Question Answering App")
     st.write("Upload an image and ask a question to get answers!")
-
+    
+    oracle = pipeline(model="dandelin/vilt-b32-finetuned-vqa")
+    
     # File uploader for image
     uploaded_image = st.file_uploader("Upload Image", type=["jpg", "jpeg", "png"])
-
+    
     if uploaded_image is not None:
         image = Image.open(uploaded_image)
         st.image(image, caption="Uploaded Image", use_column_width=True)
-
+        response = oracle(image, "What kind of food is this?")
+        st.write(f"{response}")
         # # Ask question
         # question = st.text_input("Ask your question here:")