Update prediction.py

prediction.py

@@ -40,7 +40,7 @@ def Display_prediction(model_choice, label_dict,):
         cropped_image = models.tight_center_crop(image, crop_ratio=0.7)
 
         # Show cropped image to user
-        st.image(cropped_image, caption="🧠 Tightly Center-Cropped Image", use_container_width=True)
+        st.image(cropped_image, caption="🧠 Tightly Center-Cropped Image")
 
 
         predict = False