Update app.py

app.py

@@ -5,7 +5,7 @@ from logic import *
 model, processor= load_Model()
 
 def load_image(img):
-    im = Image.open(img)
+    im = Image.open(img).convert("RGB")
     return im
 
 st.title('Ask Image Questions UDOP')
@@ -15,6 +15,15 @@ if uploaded_file is not None:
     img = load_image(uploaded_file)
     st.image(img)
     st.write("Image Uploaded Successfully")
+    with st.form("my_form"):
+        user_query = st.text_input("Ask the image")
+              
+        new_submitted = st.form_submit_button("Submit")
+        if new_submitted:
+            encoding = processor(images=img, text=user_query, return_tensors="pt")
+            outputs = model.generate(**encoding, max_new_tokens=20)
+            generated_text = processor.batch_decode(outputs, skip_special_tokens=True)[0]
+            st.write(generated_text)