Update app.py

app.py

@@ -131,33 +131,32 @@ def sample_images():
     st.header("Sample Images")
 
     # Sample images
-    sample_images = {
+    example_image = {
         "Sample 1": "img_0002.jpg",
     }
 
     # Button to load sample images
     if st.button("Load Sample Images"):
-        for sample_name, sample_path in sample_images.items():
-            st.subheader(sample_name)
-            sample_image = Image.open(sample_path).convert('RGB')
-            st.image(sample_image, caption=f"{sample_name} Image.", use_column_width=True)
-
-            # Text input for each sample image
-            text_input = st.text_area(f"Input Question for {sample_name}:")
-
-            # Predict button for each sample image
-            if st.button(f"Predict {sample_name}"):
-                if text_input:
-                    # Perform prediction
-                    prediction_result = predict(sample_image, text_input)
-
-                    # Display input text
-                    st.subheader(f"Input Question for {sample_name}:")
-                    st.write(text_input)
-
-                    # Display prediction result
-                    st.subheader(f"Prediction Result for {sample_name}:")
-                    st.write(prediction_result)
+        
+        sample_image = Image.open(example_image).convert('RGB')
+        st.image(sample_image, caption=f"{sample_name} Image.", use_column_width=True)
+
+        # Text input for each sample image
+        text_input = st.text_area(f"Input Question for {sample_name}:")
+
+        # Predict button for each sample image
+        if st.button(f"Predict"):
+            if text_input:
+                # Perform prediction
+                prediction_result = predict(sample_image, text_input)
+
+                # Display input text
+                st.subheader(f"Input Question for {"Example Image"}:")
+                st.write(text_input)
+
+                # Display prediction result
+                st.subheader(f"Prediction Result for {"Example Image"}:")
+                st.write(prediction_result)
 
 def upload_image():
     st.header("Upload Image")