Update app.py

app.py

@@ -11,7 +11,7 @@ image_processor = AutoImageProcessor.from_pretrained(repo_name)
 model = AutoModelForImageClassification.from_pretrained(repo_name)
 
 # Streamlit UI
-st.title("Image Classification with Streamlit")
+st.title("Bottomwear Image Classification with Streamlit")
 
 # Option to select image input method
 method = st.radio("Select input method", ("Upload Image", "Paste Image URL"))
@@ -30,7 +30,8 @@ elif method == "Paste Image URL":
             image = Image.open(BytesIO(response.content))
 
 if image:
-    st.image(image, caption="Uploaded Image", use_column_width=True)
+    resized_image = image.resize((500, 300))
+    st.image(resized_image, caption="Uploaded Image", use_column_width=True)
     st.write("")
     st.write("Classifying...")