ravali-maddela commited on
Commit
fdd5b20
1 Parent(s): cabcf20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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('/content/best_knn_model.joblib')
21
  else:
22
- model = joblib.load('/content/best_logistic_model.joblib')
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