Fix resolution v2

app.py

@@ -66,8 +66,10 @@ def greet(image, audio):
     seg = out_dict['heatmap'][0:1]
     seg_image = ((1 - seg.squeeze().detach().cpu().numpy()) * 255).astype(np.uint8)
     seg_image = Image.fromarray(seg_image)
+
+    original_image = image.resize((352, 352), Image.BICUBIC)
     heatmap_image = cv2.applyColorMap(np.array(seg_image), cv2.COLORMAP_JET)
-    overlaid_image = cv2.addWeighted(np.array(image.resize(352, 352)), 0.5, heatmap_image, 0.5, 0)
+    overlaid_image = cv2.addWeighted(np.array(original_image), 0.5, heatmap_image, 0.5, 0)
 
     return overlaid_image