duchaba commited on
Commit
589cb00
·
verified ·
1 Parent(s): a21e8db

update gradio Image type

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -161,7 +161,8 @@ learn = fastai.learner.load_learner('ada_learn_skin_norm2000.pkl')
161
  learn2 = fastai.learner.load_learner('ada_learn_malben.pkl')
162
  maxi.categories = learn.dls.vocab
163
  maxi.categories2 = learn2.dls.vocab
164
- hf_image = gradio.inputs.Image(shape=(192, 192))
 
165
  hf_label = gradio.outputs.Label()
166
  intf = gradio.Interface(fn=maxi.predict_donut,
167
  inputs=hf_image,
 
161
  learn2 = fastai.learner.load_learner('ada_learn_malben.pkl')
162
  maxi.categories = learn.dls.vocab
163
  maxi.categories2 = learn2.dls.vocab
164
+ #hf_image = gradio.inputs.Image(shape=(192, 192))
165
+ hf_image = gradio.Image(type="pil")
166
  hf_label = gradio.outputs.Label()
167
  intf = gradio.Interface(fn=maxi.predict_donut,
168
  inputs=hf_image,