Update app.py

app.py

@@ -12,16 +12,17 @@ model = tf.keras.models.load_model('Task_Pokemon.keras')
 class_names = ['Aerodactyl', 'Charizard', 'Victreebel']
 
 def classify_image(image):
-    img_pil = image.resize((160, 160))  # Definieren der Größe der Bilder
-    img_array = tf.keras.preprocessing.image.img_to_array(img_pil)
-    img_array = tf.expand_dims(img_array, 0)  # Erstellen eines Batches
+    image = Image.fromarray(image.astype('uint8'), 'RGB')
+    img = image.resize((150, 150))
+    img_array = tf.keras.preprocessing.image.img_to_array(img)
+    img_array = tf.expand_dims(img_array, 0)  # Erstelle einen Batch
     predictions = model.predict(img_array)
     predicted_class = class_names[np.argmax(predictions[0])]
     confidence = np.max(predictions[0])
-    return predicted_class, confidence
+    return {predicted_class: float(confidence)}
 
-image_input = gr.inputs.Image()
-label = gr.outputs.Label(num_top_classes=3)
+image_input = gr.Image()
+label = gr.Label(num_top_classes=3)
 
 iface = gr.Interface(
     fn=classify_image,
@@ -32,3 +33,5 @@ iface = gr.Interface(
 )
 
 iface.launch()
+
+