jadavis commited on
Commit
c0f7da9
·
1 Parent(s): 3156679

added input recording for debug purposes

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,11 +6,12 @@ learn=load_learner("model.pkl")
6
  labels=learn.dls.vocab
7
  def sample_predict(img):
8
  img = PILImage.create(img)
 
9
  pred,pred_idx,probs=learn.predict(img)
10
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
11
 
12
  intf=gr.Interface(fn=sample_predict,
13
  inputs=gr.components.Image(shape=(256, 256)),
14
- outputs=gr.components.Label(num_top_classes=3))
15
 
16
  intf.launch()
 
6
  labels=learn.dls.vocab
7
  def sample_predict(img):
8
  img = PILImage.create(img)
9
+ img.save("inputted/image"+"_"+str(len(os.listdir("inputted")))+".jpg")
10
  pred,pred_idx,probs=learn.predict(img)
11
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
12
 
13
  intf=gr.Interface(fn=sample_predict,
14
  inputs=gr.components.Image(shape=(256, 256)),
15
+ outputs=gr.components.Label())
16
 
17
  intf.launch()