Devon12 commited on
Commit
3d4cfbe
·
verified ·
1 Parent(s): 33e388f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("_", " ").capitalize()
98
  return label_1, tonne
99
 
100
  button.click(fn=analyze, inputs=image_input, outputs=[output_label, output_tonne])
101
 
102
- demo.launch(show_api=False)
 
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()