working

app.py

@@ -58,15 +58,17 @@ model = get_model()
 labels = ["zero","one","two","three","four","five","six","seven","eight","nine","ten","eleven","twelve","thrteen","fourteen","fifteen","sixteen","seventeen","eightteen","nineteen"]
 
 def predict(img):
-    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-    resized = cv2.resize(gray, (64, 64))
-    reshaped = resized.reshape(1, 64, 64, 1)
-    out = model.predict(reshaped)
-    cls = labels[out.argmax()]
+    img = cv2.resize(img, (IMG_WIDTH, IMG_HEIGHT), interpolation=cv2.INTER_AREA)
+    img = img.astype("float32")
+    img = img.reshape(1, IMG_HEIGHT, IMG_WIDTH, 3)    
+    out = model.predict(img)
+    cls = out.argmax()
     return cls
 
 ui = gr.Interface(
 fn=predict,
 inputs=gr.Image(type="numpy"),
 outputs=gr.Textbox()
-)
+)
+
+ui.launch()