Devmurari commited on
Commit
d9c78ae
·
1 Parent(s): b787563

indentation changes

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. app.py +6 -6
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ venv/
app.py CHANGED
@@ -9,12 +9,12 @@ model = load_model("model_1.h5")
9
  labels = ["zero","one","two","three","four","five","six","seven","eight","nine","ten","eleven","twelve","thrteen","fourteen","fifteen","sixteen","seventeen","eightteen","nineteen"]
10
 
11
  def predict(img):
12
- gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
13
- resized = cv2.resize(gray, (64, 64))
14
- reshaped = resized.reshape(1, 64, 64, 1)
15
- out = model.predict(reshaped)
16
- cls = labels[out.argmax()]
17
- return cls
18
 
19
  ui = gr.Interface(
20
  fn=predict,
 
9
  labels = ["zero","one","two","three","four","five","six","seven","eight","nine","ten","eleven","twelve","thrteen","fourteen","fifteen","sixteen","seventeen","eightteen","nineteen"]
10
 
11
  def predict(img):
12
+ gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
13
+ resized = cv2.resize(gray, (64, 64))
14
+ reshaped = resized.reshape(1, 64, 64, 1)
15
+ out = model.predict(reshaped)
16
+ cls = labels[out.argmax()]
17
+ return cls
18
 
19
  ui = gr.Interface(
20
  fn=predict,