Kupredom commited on
Commit
b81cc70
1 Parent(s): 2edbeeb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def predict_apple(image):
28
  p_zederapfel = prediction[0][2] # Probability for "Zederapfel"
29
  p_gesund = prediction[0][3] # Probability for "Gesund"
30
 
31
- return {'gesund': p_gesund, 'schorf': p_schorf, 'schwarzfaeule': p_schwarzfaeule, 'zederapfel': p_zederapfel}
32
 
33
 
34
  # Create the Gradio interface
 
28
  p_zederapfel = prediction[0][2] # Probability for "Zederapfel"
29
  p_gesund = prediction[0][3] # Probability for "Gesund"
30
 
31
+ return {'Gesund': p_gesund, 'Schorf': p_schorf, 'Schwarzfaeule': p_schwarzfaeule, 'Zederapfel': p_zederapfel}
32
 
33
 
34
  # Create the Gradio interface