Update app.py

app.py

@@ -109,12 +109,6 @@ def convert_to_base64(pil_image):
     base64_image = processing_utils.encode_pil_to_base64(pil_image)
     return base64_image
 
-def tensor_to_pil_image(tensor):
-    # Convert the tensor to a PIL Image
-    tensor = tensor.squeeze().detach().cpu()
-    tensor = transforms.ToPILImage()(tensor)
-    return tensor
-
 def inference(
     control_image: Image.Image,
     prompt: str,
@@ -187,8 +181,8 @@ def inference(
         print(f"Inference ended at {end_time_formatted}, taking {end_time-start_time}s")
         logger.debug("Output Types: generated_image=%s", type(None))
         logger.debug("Output Values: generated_image=None")
-        print(out_image["images"][0].shape)
-        return tensor_to_pil_image(out_image["images"][0])
+        print("Type of output image:", type(out_image["images"][0]))
+        return out_image["images"][0]
         
     
     except Exception as e: