Update app.py

app.py

@@ -118,6 +118,8 @@ def run(bucket_folder, person_name, req_id):
         if isinstance(img, torch.Tensor):
             img = img.detach().cpu()
             img = transforms.ToPILImage()(img)
+        elif isinstance(img, np.ndarray):
+            img = Image.fromarray(img)
         elif not isinstance(img, Image.Image):
             continue  # skip unknown formats
         temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")