Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = ['
|
| 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
|