c

app.py

@@ -98,7 +98,9 @@ def upscale(image, prompt):
 def upscale2(image, prompt):
     print("upscale2",image,prompt)
 
-    pipeline = LDMSuperResolutionPipeline.from_pretrained("CompVis/ldm-super-resolution-4x-openimages", torch_dtype=torch.float16).to('cuda')
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    pipeline = LDMSuperResolutionPipeline.from_pretrained("CompVis/ldm-super-resolution-4x-openimages")
+    pipeline = pipeline.to(device)
 
     upscaled_image = pipeline(image, num_inference_steps=10, eta=1).images[0]
     return upscaled_image
@@ -106,8 +108,6 @@ def upscale2(image, prompt):
 
 
 
-
-
 with gr.Blocks() as app:    
     with gr.Row():