Update app.py - add comment for categories and remove is_pneumonia

app.py

@@ -6,17 +6,13 @@ from fastai.vision.all import *
 import gradio as gr
 
 
-# Function to determine if an image contains pneumonia
-# Checks if the filename contains 'virus' or 'bacteria'
-def is_pneumonia(x): 
-    return (x.find('virus')!=-1 or x.find('bacteria')!=-1)
-
-
 # Load the trained fastai model for predictions
 learn = load_learner('pneumonia_model.pkl')
 
 
 # Define the possible categories for prediction
+# Use the categories directly from the DataLoaders vocab
+# categories: 'PNEUMONIA' or 'NORMAL'
 categories = learn.dls.vocab