Update app.py

app.py

@@ -123,6 +123,58 @@ def predict_upload(image):
   k=show("geeks.jpg")
   return k
 
+import numpy as np
+import gradio as gr
+
+def predict_sketch(img):
+  img_3d=img.reshape(-1,28,28,1)
+  im_resize=img_3d/255.0
+  prediction=nn.predict(im_resize).tolist()[0]
+  return {str(i):prediction[i] for i in range(10)}
+
+def predict_upload(image):
+  im1 = image.save("/content/geeks.jpg")
+  k=show("/content/geeks.jpg")
+  return k
+
+description='''
+<h1>Digit Recognizer</h4>
+<h2>You have Two options either you can draw the digit in the sketchpad or you can upload the image and see the results</h2>
+'''
+
+des2="Note:Draw Digits from 0-9 and Try to Draw the Digit in the center for better accuracy"
+
+with gr.Blocks() as demo:
+
+    gr.Markdown(description)
+    with gr.Tabs():
+        with gr.TabItem("Sketch"):
+          gr.Markdown(des2)
+          with gr.Row():
+            text_input = gr.Sketchpad()
+            text_output = gr.Label(num_top_classes=3)
+          clr = gr.Button("Submit")
+
+        with gr.TabItem("Upload Image"):
+            with gr.Row():
+                image_input =  gr.Image(type="pil",)
+                image_output =  gr.Image(type="pil",)
+            image_button = gr.Button("Submit")
+            examples=["/content/tt.jpg","/content/multitest.jpg"]
+            gr.Examples(examples=examples,
+                        inputs=image_input,
+                        fn=predict_upload,
+                        outputs=image_output)
+
+
+    image_input.change(predict_upload,image_input, image_output,examples)
+    text_input.change(predict_sketch,text_input,outputs=text_output)
+
+
+demo.launch(debug=True)
+
+
+'''
 with gr.Blocks(title="Digit Recognizer") as demo:
     gr.Markdown("Flip text or image files using this demo.")
     with gr.Tabs():
@@ -139,4 +191,5 @@ with gr.Blocks(title="Digit Recognizer") as demo:
     image_button.click(predict_upload,image_input, image_output)
     text_button.click(predict_sketch,text_input,outputs=text_output)
 
-demo.launch(debug=True)
+demo.launch(debug=True)
+'''