whichlight commited on
Commit
bd25240
1 Parent(s): 79e703c

updating specs to see if bug goes away

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +2 -2
  3. requirements.txt +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🏢
4
  colorFrom: gray
5
  colorTo: purple
6
  sdk: gradio
7
- sdk_version: 3.16.1
8
  app_file: app.py
9
  pinned: false
10
  ---
 
4
  colorFrom: gray
5
  colorTo: purple
6
  sdk: gradio
7
+ sdk_version: 3.15.0
8
  app_file: app.py
9
  pinned: false
10
  ---
app.py CHANGED
@@ -31,14 +31,14 @@ learn = load_learner('door_model.pkl')
31
  labels = learn.dls.vocab
32
 
33
  def predict(img):
34
- #img = PILImage.create(img)
35
  pred,pred_idx,probs = learn.predict(img)
36
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
37
 
38
 
39
  iface = gr.Interface(
40
  fn=predict,
41
- inputs=gr.inputs.Image(shape=(192, 192)),
42
  outputs=gr.outputs.Label(),
43
  title=title,
44
  description=description,
 
31
  labels = learn.dls.vocab
32
 
33
  def predict(img):
34
+ img = PILImage.create(img)
35
  pred,pred_idx,probs = learn.predict(img)
36
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
37
 
38
 
39
  iface = gr.Interface(
40
  fn=predict,
41
+ inputs=gr.inputs.Image(shape=(224, 224)),
42
  outputs=gr.outputs.Label(),
43
  title=title,
44
  description=description,
requirements.txt CHANGED
@@ -1,2 +1,2 @@
1
- fastai
2
  scikit-image
 
1
+ fastai==2.7.7
2
  scikit-image