Update app.py

app.py

@@ -10,36 +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:
+         clicked = st.form_submit_button("Submit")
+         if clicked:
+            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()