Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -94,9 +94,9 @@ with gr.Blocks(title="Abfallerkennung mit KI 🗑️") as demo:
|
|
94 |
_, predicted = torch.max(outputs, 1)
|
95 |
label = class_labels[predicted.item()]
|
96 |
tonne = class_to_tonne.get(label, "Unbekannt")
|
97 |
-
label_1 = label.replace("_", " ")
|
98 |
return label_1, tonne
|
99 |
|
100 |
button.click(fn=analyze, inputs=image_input, outputs=[output_label, output_tonne])
|
101 |
|
102 |
-
demo.launch(
|
|
|
94 |
_, predicted = torch.max(outputs, 1)
|
95 |
label = class_labels[predicted.item()]
|
96 |
tonne = class_to_tonne.get(label, "Unbekannt")
|
97 |
+
label_1 = label.replace("_", " ")
|
98 |
return label_1, tonne
|
99 |
|
100 |
button.click(fn=analyze, inputs=image_input, outputs=[output_label, output_tonne])
|
101 |
|
102 |
+
demo.launch()
|