Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,7 +33,7 @@ def run():
|
|
| 33 |
predictions = model.predict(img_array)
|
| 34 |
|
| 35 |
#Get class with the highest possibility
|
| 36 |
-
idx = np.where(predictions
|
| 37 |
|
| 38 |
type = ['oily', 'dry', 'normal']
|
| 39 |
result = f'Prediction: {type[idx]}'
|
|
|
|
| 33 |
predictions = model.predict(img_array)
|
| 34 |
|
| 35 |
#Get class with the highest possibility
|
| 36 |
+
idx = np.where(predictions > 0.5, 1, 0).item()
|
| 37 |
|
| 38 |
type = ['oily', 'dry', 'normal']
|
| 39 |
result = f'Prediction: {type[idx]}'
|