Update app.py

app.py

@@ -6,7 +6,7 @@ import skimage
 learn = load_learner('export.pkl')
 
 # Define the labels (in this case, just 'cat' and 'dog')
-labels = ['cat', 'dog']
+labels = learn.dls.vocab
 
 # Define the prediction function
 def predict(img):
@@ -19,7 +19,7 @@ def predict(img):
 title = "Cat vs Dog Classifier"
 description = "A classifier to distinguish between cats and dogs. Trained with fastai on a relevant dataset."
 article = "<p style='text-align: center'><a href='https://yourlinkhere.com' target='_blank'>Blog post or additional information</a></p>"
-examples = ['/path/to/example_image.jpg'] # Update this path to your example images
+examples = ['example_image.jpg'] # Update this path to your example images
 
 # Create and launch the Gradio interface
 gr.Interface(