Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 #import keras
 
 def get_digit(digit_choice):
-  if(digit_choice == '1'):
+  if(digit_choice == 1):
     return 'one'
   else:
     return 'not one'
@@ -13,9 +13,9 @@ gr.Interface(
   #inputs='image',
   #inputs=[gr.inputs.Image(label="Input Image", source="webcam")],
   inputs = [ 
-    #gr.inputs.Dropdown(['1', '2', '3'], type='index'),
+    gr.inputs.Dropdown([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]), #type='index'),
     #gr.inputs.Number()
-    'text'
+    #'text'
   ],
   #outputs='image',
   outputs='text',