Update app.py

app.py

@@ -73,8 +73,23 @@ model.fc = nn.Linear(num_of_features, num_of_classes)
 model.load_state_dict(torch.load('resnet18_model (1).pth', map_location=torch.device('cpu')))
 model.eval()
 
+st.markdown(
+    """
+    <div style='
+        font-weight: bold;
+        font-size: 20px;
+        color: #FF69B4;
+        margin-bottom: 10px;
+        text-align: center;
+        text-shadow: 1px 1px #ccc;
+    '>
+        📤 Upload an Image
+    </div>
+    """,
+    unsafe_allow_html=True
+)
 # Image upload
-uploaded_img = st.file_uploader("**Upload an image**", type=["jpg", "jpeg", "png"])
+uploaded_img = st.file_uploader("", type=["jpg", "jpeg", "png"])
 if st.button("Submit"):
     if uploaded_img is not None:
         # Display uploaded image in a smaller size
@@ -87,7 +102,7 @@ if st.button("Submit"):
             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)