Update app.py

app.py

@@ -22,14 +22,14 @@ def predict(img):
     boxes, _ = mtcnn.detect(img)
     #o_img = img.copy()
     #draw = ImageDraw.Draw(o_img)
-    draw = Image.new("RGBA", img.size, color = "white")
-    img1 = ImageDraw.Draw(img)
+    o_img = Image.new("RGBA", img.size, color = "white")
+    draw = ImageDraw.Draw(o_img)
     for box in boxes:
         coords = tuple(box.tolist())
         pred,pred_idx,probs = learn.predict(img.crop(coords))
-        img1.rectangle(coords, fill=e_colors[pred], outline=(0, 0, 0), width=1)
-        img1.text((coords[0]-10, coords[1]-10), pred, font=ImageFont.truetype("arial.ttf"))
-    return draw
+        draw.rectangle(coords, fill=e_colors[pred], outline=(0, 0, 0), width=1)
+        draw.text((coords[0]-10, coords[1]-10), pred, font=ImageFont.truetype("arial.ttf"))
+    return o_img
     '''
         emotions["x"].append((coords[0] + coords[2])/2)
         emotions["y"].append((coords[1] + coords[3])/2)