Update app.py

app.py

@@ -17,21 +17,28 @@ labels = ['Abra', 'Cloyster', 'Dodrio']
 
 # Vorhersagefunktion
 def predict(image):
-    # Bildvorverarbeitung
-    image = image.resize((64, 64))
-    image = np.array(image) / 255.0
-    image = np.expand_dims(image, axis=0)
-    predictions = model.predict(image)
-    confidences = {labels[i]: float(predictions[0][i]) for i in range(len(labels))}
-    return confidences
+    try:
+        # Bildvorverarbeitung
+        image = image.resize((64, 64))
+        image = np.array(image) / 255.0
+        image = np.expand_dims(image, axis=0)
+        
+        # Vorhersage
+        predictions = model.predict(image)
+        confidences = {labels[i]: float(predictions[0][i]) for i in range(len(labels))}
+        
+        return confidences
+    except Exception as e:
+        return str(e)  # Fehlernachricht zurückgeben
 
 # Gradio-Interface erstellen
 iface = gr.Interface(
     fn=predict, 
-    inputs=gr.Image(),  # Keine shape-Parameter hier
+    inputs=gr.Image(type="pil"),  # Bild als PIL-Objekt
     outputs=gr.Label(),
     description="Pokémon Classifier"
 )
 
 if __name__ == "__main__":
     iface.launch()
+