guru001 commited on
Commit
a37501a
1 Parent(s): 5ec701f

should work now

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ def sign(input_img):
8
  input_img = Image.fromarray(input_img)
9
  prediction = sign_api.predict(input_img)
10
  print('prediction',prediction)
11
- return prediction
12
 
13
  demo = gr.Interface(sign,inputs=gr.Image(shape=(200, 200)), outputs=gr.Label())
14
  demo.launch()
 
8
  input_img = Image.fromarray(input_img)
9
  prediction = sign_api.predict(input_img)
10
  print('prediction',prediction)
11
+ return prediction['class']
12
 
13
  demo = gr.Interface(sign,inputs=gr.Image(shape=(200, 200)), outputs=gr.Label())
14
  demo.launch()