ravali-maddela
commited on
Commit
•
fdd5b20
1
Parent(s):
cabcf20
Update app.py
Browse files
app.py
CHANGED
@@ -17,9 +17,9 @@ def fashion_images(input1, input2):
|
|
17 |
print("numpy_image: ", numpy_image.shape)
|
18 |
|
19 |
if input2 == "KNN":
|
20 |
-
model = joblib.load('
|
21 |
else:
|
22 |
-
model = joblib.load('
|
23 |
out = model.predict(numpy_image)[0]
|
24 |
out_prob = model.predict_proba(numpy_image)[0]
|
25 |
|
|
|
17 |
print("numpy_image: ", numpy_image.shape)
|
18 |
|
19 |
if input2 == "KNN":
|
20 |
+
model = joblib.load('best_knn_model.joblib')
|
21 |
else:
|
22 |
+
model = joblib.load('best_logistic_model.joblib')
|
23 |
out = model.predict(numpy_image)[0]
|
24 |
out_prob = model.predict_proba(numpy_image)[0]
|
25 |
|