Update app.py

app.py

@@ -13,46 +13,22 @@ def predict(image):
     guess1, guess2 = cal_est[0]['answer'], cal_est[1]['answer']
     return f"This is {type_food[0]['answer']}. I estimate this to contain {min(guess1, guess2)}-{max(guess1, guess2)} calories"
 oracle = pipeline(model="dandelin/vilt-b32-finetuned-vqa")
-css = '''
-h1#title {
-  text-align: center;
-}
-h3#header {
-  text-align: center;
-}
-img#overview {
-  max-width: 800px;
-  max-height: 600px;
-}
-img#style-image {
-  max-width: 1000px;
-  max-height: 600px;
-}
-'''
-demo = gr.Blocks(css=css)
-with demo:
-  gr.Markdown('''<h1 id="title">Image Caption 🖼️</h1>''')
-  gr.Markdown('''Made by : ''')
-  with gr.Column():
-        input = gr.inputs.Image(label="Upload your Image", type = 'pil', optional=True)
-        output = gr.outputs.Textbox(type="auto",label="Captions")
-  btn = gr.Button("Genrate Caption")
-  btn.click(fn=predict, inputs=input, outputs=output)
-demo.launch()
-# def main():
-#     st.title("Image Question Answering App")
-#     st.write("Upload an image and ask a question to get answers!")
+
+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")
+    oracle = pipeline(model="dandelin/vilt-b32-finetuned-vqa")
 
-#     while True:
-#         # File uploader for image
-#         uploaded_image = st.file_uploader("Upload Image", type=["jpg", "jpeg", "png"])
+    while True:
+        # 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)
-            
+        if uploaded_image is not None:
+            image = Image.open(uploaded_image)
+            st.image(image, caption="Uploaded Image", use_column_width=True)
+            response = predict(image)
+            st.write(response)
 
-# if __name__ == "__main__":
-#     main()
+if __name__ == "__main__":
+    main()