Update app.py

app.py

@@ -1,12 +1,11 @@
+from pathlib import Path
 
-pip3 install torch
 import torch
+import gradio as gr
 from torch import nn
-from pathlib import Path
 
 
-
-LABELS=Path('class_names.txt').read_text().splitlines()
+LABELS = Path('class_names.txt').read_text().splitlines()
 
 model = nn.Sequential(
     nn.Conv2d(1, 32, 3, padding='same'),
@@ -23,11 +22,7 @@ model = nn.Sequential(
     nn.ReLU(),
     nn.Linear(256, len(LABELS)),
 )
-
-state_dict=torch.load('pytorch_model.bin',map_location='cpu')
-
-import gradio as gr
-
+state_dict = torch.load('pytorch_model.bin', map_location='cpu')
 model.load_state_dict(state_dict, strict=False)
 model.eval()
 
@@ -43,14 +38,13 @@ def predict(im):
 
     return {LABELS[i]: v.item() for i, v in zip(indices, values)}
 
-interface=gr.Interface(
-    predict,
-    inputs="sketchpad",
-    outputs='label',
-    theme="huggingface",
-    title="Sketch Recognition",
-    description="Sketch something for the model to guess in realtime",
+interface = gr.Interface(
+    predict, 
+    inputs="sketchpad", 
+    outputs='label', 
+    theme="huggingface", 
+    title="Sketch Recognition", 
+    description="Who wants to play Pictionary? Draw a common object like a shovel or a laptop, and the algorithm will guess in real time!", 
     article = "<p style='text-align: center'>Sketch Recognition | Demo Model</p>",
     live=True)
-interface.launch(share=True)
-
+interface.launch(debug=True)