Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -52,7 +52,7 @@ def get_outfit_recommendation(pred_class):
|
|
| 52 |
elif pred_class == 'Bag':
|
| 53 |
return 'Top'
|
| 54 |
else:
|
| 55 |
-
return '
|
| 56 |
|
| 57 |
def predict(image):
|
| 58 |
pred_class, pred_idx, outputs = learn.predict(image)
|
|
|
|
| 52 |
elif pred_class == 'Bag':
|
| 53 |
return 'Top'
|
| 54 |
else:
|
| 55 |
+
return 'Item'
|
| 56 |
|
| 57 |
def predict(image):
|
| 58 |
pred_class, pred_idx, outputs = learn.predict(image)
|