Update app.py

app.py

@@ -99,7 +99,7 @@ def txt_to_img(
     pipe = MODELS[model_name].pipe_t2i
 
     if torch.cuda.is_available():
-        pipe = pipe.to("cuda")
+        pipe = pipe.to("cpu")
         pipe.enable_xformers_memory_efficient_attention()
 
     low_res_latents = pipe(
@@ -216,7 +216,7 @@ with gr.Blocks(css="style.css") as demo:
     with gr.Column(scale=100):
         with gr.Row():
             with gr.Column(scale=75):
-                up_res_image = gr.Image(label="Upscaled 1024px Image", shape=(1024, 1024))
+                up_res_image = gr.Image(label="Upscaled 1024px Image", shape=(1024, 512))
             with gr.Column(scale=25):
                 low_res_image = gr.Image(label="Original 512px Image", shape=(512, 512))
         error_output = gr.Markdown()