enhanced UI

src/streamlit_app.py

@@ -86,12 +86,16 @@ with tab2:
         st.image(image, caption="Uploaded Image", width=300)
 
         input_tensor = preprocess_image(image)
-        with torch.no_grad():
-            outputs = model(input_tensor)
-            _, predicted = torch.max(outputs, 1)
-            predicted_label = CLASS_NAMES[predicted.item()]
-            species_name = LABEL_MAP[predicted_label]
-            st.success(f"Predicted Monkey Species: {species_name}")
+
+        # Add spinner to indicate loading
+        with st.spinner("Classifying... Please wait."):
+            with torch.no_grad():
+                outputs = model(input_tensor)
+                _, predicted = torch.max(outputs, 1)
+                predicted_label = CLASS_NAMES[predicted.item()]
+                species_name = LABEL_MAP[predicted_label]
+
+        st.success(f"Predicted Monkey Species: {species_name}")
 
 # Tab 3: Model details and performance
 with tab3: