added txt

app.py

@@ -41,7 +41,7 @@ def get_y(item):
     return class_names[item['labels']]  # Map label index to class name
 
 # Load the model
-learn = load_learner('poke_model.pkl', cpu=False)  # Ensure correct path to poke_model.pkl
+learn = load_learner('poke_model.pkl')
 
 # Categories are derived from the vocabulary of the dataloader used during training
 categories = learn.dls.vocab  # These are the class labels used in the trained model