up test 1

app.py

@@ -71,8 +71,10 @@ def segmentation(img):
 def upscale(image,prompt):
     print("upscale")
 
-    # pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")
-    pipeline = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16).to('cuda')
+    image = Image.thumbnail((512, 512))
+
+    pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")
+    # pipeline = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16).to('cuda')
     upscaled_image = pipeline(prompt=prompt, image=image).images[0]
     return upscaled_image
 
@@ -99,6 +101,7 @@ with gr.Blocks() as app:
                                                 gr.Image(),
                                                 gr.Image(),
                                                 gr.Image()])
+        
         with gr.Column():  
             gr.Button("Segmentation").click(segmentation, inputs=gr.Image(type="pil"), outputs=gr.JSON())