Update app.py

app.py

@@ -38,15 +38,16 @@ elif option == "Image Classification":
 
     uploaded_file = st.file_uploader("Choose an image file", type=["jpg", "jpeg", "png"])
     text = st.text_area("Enter possible class names (comma-separated)")
-    if uploaded_file is not None and text:
-        candidate_labels = [t.strip() for t in text.split(',')]
-        image = Image.open(uploaded_file)
-        st.image(image, caption="Uploaded Image", use_column_width=True)
-        classification_result = imgpipe(image, candidate_labels)
-        for result in classification_result:
-            st.write(f"Label: {result['label']}, Score: {result['score']}")
-    else:
-        st.write("Please upload an image file and enter class names.")
+    if st.button("Submit"):
+        if uploaded_file is not None and text:
+            candidate_labels = [t.strip() for t in text.split(',')]
+            image = Image.open(uploaded_file)
+            st.image(image, caption="Uploaded Image", use_column_width=True)
+            classification_result = imgpipe(image, candidate_labels)
+            for result in classification_result:
+                st.write(f"Label: {result['label']}, Score: {result['score']}")
+        else:
+            st.write("Please upload an image file and enter class names.")
         
 
 else: