Adjust output resolution

app.py

@@ -69,9 +69,9 @@ def greet(image, audio):
     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)
+    seg_image = seg_image.resize(image.size, Image.BICUBIC)
     heatmap_image = cv2.applyColorMap(np.array(seg_image), cv2.COLORMAP_JET)
-    overlaid_image = cv2.addWeighted(np.array(original_image), 0.5, heatmap_image, 0.5, 0)
+    overlaid_image = cv2.addWeighted(np.array(image), 0.5, heatmap_image, 0.5, 0)
 
     return overlaid_image