Update app.py

app.py

@@ -90,18 +90,14 @@ def predict_image(image):
         _, predicted = torch.max(outputs, 1)
         label = class_labels[predicted.item()]
         tonne = class_to_tonne.get(label, "Unbekannt")
-    
-    return {
-        "Klasse (auf Deutsch)": label.replace("_", " ").capitalize(),
-        "Gehört in die": tonne
-    }
+    return label.replace("_", " ").capitalize(), tonne
 
 # Gradio interface setup
 interface = gr.Interface(
     fn=predict_image,
     inputs=gr.Image(type="pil", label="Bild hochladen"),
     outputs=[
-        gr.Textbox(label="Vorhersage (Klasse auf Deutsch)"),
+        gr.Textbox(label="Vorhersage"),
         gr.Textbox(label="Richtiger Abfallbehälter")
     ],
     title="Abfallerkennung mit KI 🗑️",