Spaces:
Sleeping
Sleeping
image size
Browse files
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=(
|
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)
|