Spaces:

Mosharof
/

Digit_Recognition

Runtime error

Mosharof commited on Oct 15, 2023

Commit

eeb4410

1 Parent(s): 3da089f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ learn = load_learner('model.pkl')
 def classify_image(img):
     pred, idx,probs = learn.predict(img)
-    return pred, idx,probs
 image = gr.inputs.Image(shape=(192,192))
 lebel = gr.outputs.Label()

 def classify_image(img):
     pred, idx,probs = learn.predict(img)
+    return pred
 image = gr.inputs.Image(shape=(192,192))
 lebel = gr.outputs.Label()