adds title and description

app.py

@@ -6,6 +6,9 @@ with open('model_91_7248.bin', 'rb') as f:
 
 
 def predict(input):
+    if input is None:
+        return 'None'
+
     x = input.reshape((784, 1))
     p = nn.feed_forward(x).reshape((10,))
 
@@ -14,6 +17,8 @@ def predict(input):
 
 demo = gr.Interface(
     fn=predict,
+    title='Simple NeuralNet for handwritten digits classification',
+    description='For a more robust model check this out: <a href="https://huggingface.co/spaces/ffcm/cnn-pytorch-mnist">ConvNet for handwritten digits classification</a>',
     inputs=[
         gr.Sketchpad(
             shape=(28, 28),
@@ -23,10 +28,9 @@ demo = gr.Interface(
     outputs=[
         gr.Label(
             num_top_classes=3,
-            scale=2,
+            scale=3,
         )
     ],
     live=True,
     allow_flagging=False,
-
 ).launch()