balakrish181 commited on
Commit
68c1bed
·
1 Parent(s): 34b3108

modified doodle

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,9 +20,9 @@ def predict(img):
20
 
21
  title = 'MNIST Digit Prediction'
22
  description = 'Predict handwritten digits (0-9) using a trained model.'
23
- inputs = gr.Image(type='pil', label='Upload an image of a digit')
24
  outputs = gr.Label(num_top_classes=3, label='Predictions')
25
  demo = gr.Interface(fn=predict, inputs=inputs, outputs=outputs, title=title, description=description)
26
 
27
  # Launch the interface
28
- demo.launch()
 
20
 
21
  title = 'MNIST Digit Prediction'
22
  description = 'Predict handwritten digits (0-9) using a trained model.'
23
+ inputs = gr.Sketchpad(label='Draw a digit')
24
  outputs = gr.Label(num_top_classes=3, label='Predictions')
25
  demo = gr.Interface(fn=predict, inputs=inputs, outputs=outputs, title=title, description=description)
26
 
27
  # Launch the interface
28
+ demo.launch(share=True)