Update app.py

app.py

@@ -9,7 +9,7 @@ from PIL import Image, ImageDraw
 device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 print('Running on device: {}'.format(device))
 
-mtcnn = MTCNN(margin=20, keep_all=True, post_process=False, device=device)
+mtcnn = MTCNN(margin=40, keep_all=True, post_process=False, device=device)
 
 learn = load_learner('export.pkl')
 
@@ -26,7 +26,7 @@ def predict(img):
         coords = tuple(box.tolist())
         pred,pred_idx,probs = learn.predict(img.crop(coords))
         draw.rectangle(coords, outline=(0, 0, 0), width=1)
-        button_draw.text((coords[0]-10, coords[1]-10), pred, font=ImageFont.truetype("arial"))
+        draw.text((coords[0]-10, coords[1]-10), pred, font=ImageFont.truetype("arial"))
     return o_img
     '''
         emotions["x"].append((coords[0] + coords[2])/2)