Update streamlit_app.py

streamlit_app.py

@@ -49,7 +49,6 @@ def load_model():
         # Fallback: assume it matches class_names
         num_classes_in_model = len(class_names)
     
-    st.info(f"Model has {num_classes_in_model} classes, class_names.txt has {len(class_names)} classes")
     
     # Create model with the correct number of classes from the saved model
     model = timm.create_model('efficientnet_b0', pretrained=False, num_classes=num_classes_in_model)