added txt

app.py

@@ -52,7 +52,7 @@ def classify_pokemon(img):
     return dict(zip(categories, map(float, probs)))
 
 # Gradio interface setup
-image = gr.Image(shape=(128, 128))  # Image size should match your training data size (e.g., 128x128)
+image = gr.Image(type='pil')  # Image size should match your training data size (e.g., 128x128)
 label = gr.Label()
 
 # Set up Gradio interface