heinini2 commited on
Commit
1b04969
1 Parent(s): 5032fa6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def predict_pokemon(image):
21
 
22
  # Bild in das Modell einspeisen und Vorhersage treffen
23
  prediction = model.predict(image[None, ...])
24
- confidences = {labels[i]: np.round(float(prediction[0][i]), 2) for i in range(len(labels))} return confidences
25
  return confidences
26
 
27
  # Gradio Interface definieren
 
21
 
22
  # Bild in das Modell einspeisen und Vorhersage treffen
23
  prediction = model.predict(image[None, ...])
24
+ confidences = {labels[i]: np.round(float(prediction[0][i]), 2) for i in range(len(labels))}
25
  return confidences
26
 
27
  # Gradio Interface definieren