tobiaserni1 commited on
Commit
f08cd20
·
verified ·
1 Parent(s): de55aee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def predict_blumen(img):
16
  img_array = preprocess_input(img_array) # Preprocess the input as expected by ResNet50
17
 
18
  prediction = model.predict(img_array) # Predict using the model
19
- classes = ['common_daisy', 'sunflower', 'tulip' ] # Specific flower names
20
  return {classes[i]: float(prediction[0][i]) for i in range(3)} # Return the prediction
21
 
22
  # Define Gradio interface
 
16
  img_array = preprocess_input(img_array) # Preprocess the input as expected by ResNet50
17
 
18
  prediction = model.predict(img_array) # Predict using the model
19
+ classes = ['astilbe', 'sunflower', 'tulip' ] # Specific flower names
20
  return {classes[i]: float(prediction[0][i]) for i in range(3)} # Return the prediction
21
 
22
  # Define Gradio interface