Update app.py

app.py

@@ -10,35 +10,36 @@ def main():
 
     with st.form("text_field"):
         uploaded_file = st.file_uploader("Upload Files",type=['png','jpeg','jpg'])
+     clicked = st.form_submit_button("Submit")
+     if clicked:
+        if uploaded_file!=None:
     
-    if uploaded_file!=None:
-
-        img=Image.open(uploaded_file)
-
-        extractor = AutoFeatureExtractor.from_pretrained("yangy50/garbage-classification")
-        model = AutoModelForImageClassification.from_pretrained("yangy50/garbage-classification")
-
-        inputs = extractor(img,return_tensors="pt")
-        outputs = model(**inputs)
-        label_num=outputs.logits.softmax(1).argmax(1)
-        label_num=label_num.item()
-
-        st.write("The prediction class is:")
-
-        if label_num==0:
-            st.write("cardboard")
-        elif label_num==1:
-            st.write("glass")
-        elif label_num==2:
-            st.write("metal")
-        elif label_num==3:
-            st.write("paper")
-        elif label_num==4:
-            st.write("plastic")
-        else:
-            st.write("trash")
-
-        st.image(img)
+            img=Image.open(uploaded_file)
+    
+            extractor = AutoFeatureExtractor.from_pretrained("yangy50/garbage-classification")
+            model = AutoModelForImageClassification.from_pretrained("yangy50/garbage-classification")
+    
+            inputs = extractor(img,return_tensors="pt")
+            outputs = model(**inputs)
+            label_num=outputs.logits.softmax(1).argmax(1)
+            label_num=label_num.item()
+    
+            st.write("The prediction class is:")
+    
+            if label_num==0:
+                st.write("cardboard")
+            elif label_num==1:
+                st.write("glass")
+            elif label_num==2:
+                st.write("metal")
+            elif label_num==3:
+                st.write("paper")
+            elif label_num==4:
+                st.write("plastic")
+            else:
+                st.write("trash")
+    
+            st.image(img)
 
 if __name__ == "__main__":
     main()