dr4g0n7ly commited on
Commit
31dd99d
·
1 Parent(s): f7cb967

image size

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ def classify_img(img):
7
  pred,idx,probs = learn.predict(img)
8
  return str(pred + ", probability: " + str(round(torch.max(probs).item(), 4)) )
9
 
10
- image = gr.inputs.Image(shape=(192, 192))
11
  label = gr.outputs.Label()
12
 
13
  intf = gr.Interface(fn=classify_img, inputs=image, outputs=label)
 
7
  pred,idx,probs = learn.predict(img)
8
  return str(pred + ", probability: " + str(round(torch.max(probs).item(), 4)) )
9
 
10
+ image = gr.inputs.Image(shape=(400, 400))
11
  label = gr.outputs.Label()
12
 
13
  intf = gr.Interface(fn=classify_img, inputs=image, outputs=label)