Spaces:

rupeshs
/

fastsdcpu

Running

rupeshs commited on Nov 1, 2023

Commit

0d7ef9c

•

1 Parent(s): 7f3be1a

upscaler update

Files changed (1) hide show

frontend/webui/text_to_image_ui.py CHANGED Viewed

@@ -81,7 +81,9 @@ def generate_text_to_image(
         #out_images.append(image.resize((768, 768),resample=Image.LANCZOS))
         in_image = ImageLoader.load_image(image)
         up_image =upscaler(in_image)
-        out_images.append(up_image)
     return out_images

         #out_images.append(image.resize((768, 768),resample=Image.LANCZOS))
         in_image = ImageLoader.load_image(image)
         up_image =upscaler(in_image)
+        transform = T.ToPILImage()
+        pil_img=transform(up_image.squeeze())
+        out_images.append(pil_img)
     return out_images