CullerWhale commited on
Commit
525ba71
·
verified ·
1 Parent(s): 37ee1fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -36,10 +36,11 @@ learn = load_learner("templateClassifierDATAhalfEPOCHoneVISION.pkl")
36
  print("Model Vocabulary:", learn.dls.vocab)
37
 
38
  labels = learn.dls.vocab
39
- def predict(img):
40
- img = PILImage.create(img)
41
- pred,pred_idx,probs = learn.predict(img)
42
- return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
43
 
44
  def predict(img):
45
  img = PILImage.create(img)
 
36
  print("Model Vocabulary:", learn.dls.vocab)
37
 
38
  labels = learn.dls.vocab
39
+
40
+ # def predict(img):
41
+ # img = PILImage.create(img)
42
+ # pred,pred_idx,probs = learn.predict(img)
43
+ # return {labels[i]: float(probs[i]) for i in range(len(labels))}
44
 
45
  def predict(img):
46
  img = PILImage.create(img)