duchaba commited on
Commit
b58f82c
·
verified ·
1 Parent(s): 2308d08

update pil image

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -7,6 +7,7 @@ import matplotlib
7
  import numpy
8
  import pandas
9
  from fastai.vision.all import *
 
10
  #
11
  # create class
12
  class ADA_SKIN(object):
@@ -84,11 +85,11 @@ class ADA_SKIN(object):
84
  return
85
  #
86
  def _predict_image(self,img,cat):
87
- pred,idx,probs = learn.predict(img)
88
  return dict(zip(cat, map(float,probs)))
89
  #
90
  def _predict_image2(self,img,cat):
91
- pred,idx,probs = learn2.predict(img)
92
  return dict(zip(cat, map(float,probs)))
93
  #
94
  def _draw_pred(self,df_pred, df2):
 
7
  import numpy
8
  import pandas
9
  from fastai.vision.all import *
10
+ from fastai.vision.all import load_learner, PILImage
11
  #
12
  # create class
13
  class ADA_SKIN(object):
 
85
  return
86
  #
87
  def _predict_image(self,img,cat):
88
+ pred,idx,probs = learn.predict(PILImage.create(img))
89
  return dict(zip(cat, map(float,probs)))
90
  #
91
  def _predict_image2(self,img,cat):
92
+ pred,idx,probs = learn2.predict(PILImage.create(img))
93
  return dict(zip(cat, map(float,probs)))
94
  #
95
  def _draw_pred(self,df_pred, df2):