Update app.py

app.py

@@ -71,7 +71,7 @@ st.divider()
 
 file_name = st.file_uploader("Choose an image...")
 
-st.button("execute classification", type="primary")
+#st.button("execute classification", type="primary")
 
 
 if file_name is not None:
@@ -87,6 +87,7 @@ if file_name is not None:
     #if image.format != "JPEG" and compress :
     if  compress :
         image = jpg_compression(image)
+        st.button("re-run classification")
         
     col1.image(image, use_column_width=True)  
     predictions = get_prediction(image)