Update app.py

app.py

@@ -17,7 +17,8 @@ def classify_image(image):
     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.Image()  # Entferne den `shape` Parameter
 label = gr.Label(num_top_classes=3)