Update app.py

app.py

@@ -20,15 +20,16 @@ emotions = {"x": [], "y": [], "State": []}
 def predict(img):
     img = PILImage.create(img)
     boxes, _ = mtcnn.detect(img)
-    o_img = img.copy()
+    #o_img = img.copy()
     #draw = ImageDraw.Draw(o_img)
     draw = Image.new("RGBA", img.size, color = "white")
+    img1 = ImageDraw.Draw(img)
     for box in boxes:
         coords = tuple(box.tolist())
         pred,pred_idx,probs = learn.predict(img.crop(coords))
-        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
+        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
     '''
         emotions["x"].append((coords[0] + coords[2])/2)
         emotions["y"].append((coords[1] + coords[3])/2)