Update app.py

app.py

@@ -75,32 +75,41 @@ model.eval()
 
 # Image upload
 uploaded_img = st.file_uploader("**Upload an image**", type=["jpg", "jpeg", "png"])
-
-if uploaded_img is not None:
-    # Display uploaded image in a smaller size
-    image = Image.open(uploaded_img)
-    st.image(image, caption="**Uploaded Image**", width=200)
-
-    # Image transformations
-    sample_transform = transforms.Compose([
-        transforms.Resize((224, 224)),
-        transforms.ToTensor(),
-        transforms.Normalize(mean=[0.1776, 0.1776, 0.1776], std=[0.1735, 0.1735, 0.1735])
-    ])
-
-    # Apply transformations
-    transformed_img = sample_transform(image).unsqueeze(0)
-
-    # Model inference
-    with torch.no_grad():
-        pred = model(transformed_img).argmax(dim=1).item()
+if st.button("Submit"):
+    if uploaded_img is not None:
+        # Display uploaded image in a smaller size
+        image = Image.open(uploaded_img)
+        st.image(image, caption="**Uploaded Image**", width=200)
+    
+        # Image transformations
+        sample_transform = transforms.Compose([
+            transforms.Resize((224, 224)),
+            transforms.ToTensor(),
+            transforms.Normalize(mean=[0.1776, 0.1776, 0.1776], std=[0.1735, 0.1735, 0.1735])
+        ])
+    
+        # Apply transformations
+        transformed_img = sample_transform(image).unsqueeze(0)
     
-    # Stylish output box
-    st.markdown(
-        f"""
-        <div class="output-container">
-            🧠 <strong>Predicted Class:</strong> {class_names[pred]}
-        </div>
-        """,
-        unsafe_allow_html=True
-    )
+        # Model inference
+        with torch.no_grad():
+            pred = model(transformed_img).argmax(dim=1).item()
+        
+        # Stylish output box
+        st.markdown(
+            f"""
+            <div class="output-container">
+                🧠 <strong>Predicted Class:</strong> {class_names[pred]}
+            </div>
+            """,
+            unsafe_allow_html=True
+        )
+    else:
+        st.markdown(
+                """
+                <div style='background-color: #f8d7da; padding: 10px; border-radius: 5px;'>
+                    <h4 style='color: #721c24;'> ⚠️ Plese upload image </h4>
+                </div>
+                """,
+                unsafe_allow_html=True
+            )