Update app.py

app.py

@@ -73,5 +73,4 @@ if uploaded_file is not None:
         pred_label = reversed_map[pred_class.item()]
 
     
-    st.pyplot(fig)
     st.write(f"Predicted Class: {pred_label}")