guru001 commited on
Commit
6dc79d9
1 Parent(s): d7579e3

updated app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,10 +1,11 @@
1
  import numpy as np
2
  import gradio as gr
3
  from model import api
4
-
5
  sign_api = api()
6
 
7
  def sign(input_img):
 
8
  prediction = sign_api.predict(input_img)
9
  return prediction
10
 
 
1
  import numpy as np
2
  import gradio as gr
3
  from model import api
4
+ from PIL import Image
5
  sign_api = api()
6
 
7
  def sign(input_img):
8
+ input_img = Image.fromarray(input_img)
9
  prediction = sign_api.predict(input_img)
10
  return prediction
11