Spaces:
Running
Running
update pil image
Browse files
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):
|