Update app.py

app.py

@@ -35,6 +35,9 @@ if torch.cuda.is_available():
         use_safetensors=True,
         variant='fp16')
 
+    # LoRA for 360° images
+    pipe.unet.load_attn_procs('artificialguybr/360Redmond')
+
     if ENABLE_CPU_OFFLOAD:
         pipe.enable_model_cpu_offload()
         refiner.enable_model_cpu_offload()
@@ -46,8 +49,7 @@ if torch.cuda.is_available():
         pipe.unet = torch.compile(pipe.unet,
                                   mode='reduce-overhead',
                                   fullgraph=True)
-        
-     pipeline.unet.load_attn_procs('artificialguybr/360Redmond')
+
 else:
     pipe = None
     refiner = None