Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ models = {
|
|
| 50 |
def predict(image, model_name):
|
| 51 |
model_data = models[model_name]
|
| 52 |
|
| 53 |
-
image_array =
|
| 54 |
|
| 55 |
model = model_data["model"]
|
| 56 |
model_type = model_data["type"]
|
|
|
|
| 50 |
def predict(image, model_name):
|
| 51 |
model_data = models[model_name]
|
| 52 |
|
| 53 |
+
image_array = model_data["image_format"](image) # Formatter l'image de la façon qui correspond
|
| 54 |
|
| 55 |
model = model_data["model"]
|
| 56 |
model_type = model_data["type"]
|