Update app.py

app.py

@@ -384,7 +384,7 @@ elif uploaded_image is not None:
         input_image = Image.open(uploaded_image)
     
         # Display the image
-        st.image(input_image, caption="Uploaded Image.", use_column_width=True)
+        st.image(input_image, caption="Uploaded Image.", use_container_width=True)
     
         # Classify the image and display the result
         predictions = classifyImage(input_image)