Update app.py

app.py

@@ -68,7 +68,7 @@ def load_model():
         MODEL_NAME,
         subfolder="text_encoder",
         torch_dtype=dtype,
-    ).to(device)
+    ).to('cpu')
 
     # Tokenizer / Processor
     tokenizer = AutoTokenizer.from_pretrained(
@@ -153,8 +153,8 @@ def process_images(
                 result = pipe(
                     images[0],
                     prompt=prompt,
-                    height=1024,
-                    width=1024,
+                    height=512,
+                    width=512,
                     negative_prompt=" ",
                     num_inference_steps=num_steps,
                     true_cfg_scale=true_cfg_scale,
@@ -164,8 +164,8 @@ def process_images(
                 result = pipe(
                     images=images,
                     prompt=prompt,
-                    height=1024,
-                    width=1024,
+                    height=512,
+                    width=512,
                     negative_prompt=" ",
                     num_inference_steps=num_steps,
                     true_cfg_scale=true_cfg_scale,
@@ -194,7 +194,6 @@ with gr.Blocks(
 
 - **Model**: UniPic-3 DMD  
 - **Inference**: 8-step fast generation  
-- **GPU-only Hugging Face Space**
 """
     )
 
@@ -240,7 +239,7 @@ with gr.Blocks(
 # -----------------------------------------------------------------------------
 @GPU
 def main():
-    demo.launch()
+    demo.launch(ssr_mode=False)
 
 
 if __name__ == "__main__":