Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ local_model_path = snapshot_download(repo_id="ElBeh/ma_basemodel")
|
|
| 9 |
model = keras.models.load_model(local_model_path)
|
| 10 |
|
| 11 |
def get_prediction(img):
|
| 12 |
-
x =
|
| 13 |
x = np.expand_dims(x, axis=0)
|
| 14 |
predictions = model.predict(x)
|
| 15 |
return predictions
|
|
|
|
| 9 |
model = keras.models.load_model(local_model_path)
|
| 10 |
|
| 11 |
def get_prediction(img):
|
| 12 |
+
x = np.array(img)
|
| 13 |
x = np.expand_dims(x, axis=0)
|
| 14 |
predictions = model.predict(x)
|
| 15 |
return predictions
|