Update app.py

app.py

@@ -29,7 +29,7 @@ for model_name, path in MODEL_PATHS.items():
         trained_models[model_name] = joblib.load(path)
 
 # Define class names (modify based on dataset)
-CLASS_NAMES = train.class_names  # Update with actual labels
+CLASS_NAMES = ['ADI','DEB','LYM','MUC','MUS','NOR','STR','TUM']  # Update with actual labels
 
 # Streamlit UI
 st.title("Multi-Model Image Classifier")