added txt

app.py

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