Update app_lora1.py

app_lora1.py

@@ -179,9 +179,7 @@ def get_pipeline(script_name):
         pipe.transformer = log_pipe_calls(pipe.transformer, "pipe.transformer")
     pipe = log_pipe_calls(pipe, "pipe")
 
-    # ZeroGPU-safe
-    if not hasattr(pipe, "hf_device_map"):
-        pipe = pipe.to("cuda")
+   
 
     PIPELINES[script_name] = pipe
     log(f"✅ Pipeline ready: {script_name}")
@@ -210,6 +208,9 @@ def generate_image(
         raise RuntimeError("Pipeline not registered")
 
     pipe = get_pipeline(pipeline_name)
+    
+    if not hasattr(pipe, "hf_device_map"):
+      pipe = pipe.to("cuda")
 
     log("=== PIPELINE TECHNOLOGY ===")
     log(pipeline_technology_info(pipe))