Update app_lora1.py

app_lora1.py

@@ -178,9 +178,7 @@ def get_pipeline(script_name):
     if pipe is None:
         raise RuntimeError(f"{script_name} did not define `pipe`.")
 
-    # ZeroGPU safe
-    if not hasattr(pipe, "hf_device_map"):
-        pipe = pipe.to("cuda")
+    
 
     PIPELINES[script_name] = pipe
     log("✅ Pipeline ready")
@@ -211,6 +209,8 @@ def generate_image(
 
     log("=== PIPELINE TECHNOLOGY ===")
     log(pipeline_technology_info(pipe))
+    if not hasattr(pipe, "hf_device_map"):
+      pipe = pipe.to("cuda")
     log("=== NEW GENERATION REQUEST ===")
     log(f"Pipeline: {pipeline_name}")
     log(f"Prompt: {prompt}")