T1

app.py

@@ -19,6 +19,7 @@ import json
 from diffusers import DiffusionPipeline
 from diffusers import StableDiffusionLatentUpscalePipeline, StableDiffusionPipeline
 from diffusers import StableDiffusionUpscalePipeline
+from diffusers import LDMSuperResolutionPipeline
 
 
 def removeFurniture(input_img1,
@@ -94,6 +95,18 @@ def upscale(image, prompt):
 
     return upscaled_image
 
+def upscale2(image, prompt):
+    print("upscale2",image,prompt)
+
+    pipeline = LDMSuperResolutionPipeline.from_pretrained("CompVis/ldm-super-resolution-4x-openimages", torch_dtype=torch.float16).to('cuda')
+
+    upscaled_image = pipeline(image, num_inference_steps=10, eta=1).images[0]
+    return upscaled_image
+
+
+
+
+
 
 with gr.Blocks() as app:    
     with gr.Row():
@@ -125,6 +138,9 @@ with gr.Blocks() as app:
         with gr.Column():
             gr.Button("Upscale").click(upscale, inputs=[gr.Image(type="pil"),gr.Textbox(label="prompt",value="empty room")], outputs=gr.Image())
 
+        with gr.Column():
+            gr.Button("Upscale2").click(upscale2, inputs=[gr.Image(type="pil"),gr.Textbox(label="prompt",value="empty room")], outputs=gr.Image())
+
 
 app.launch(debug=True,share=True)