Update app.py

app.py

@@ -133,8 +133,7 @@ Now, use this detailed instruction prompt to generate the decomposed sub-prompts
 
 def init_pipeline():
     device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
-    pipe = ProT2IPipeline.from_pretrained("SG161222/RealVisXL_V4.0", use_safetensors=True, variant='fp16').to(torch.float16)
-    pipe.enable_model_cpu_offload()
+    pipe = ProT2IPipeline.from_pretrained("SG161222/RealVisXL_V4.0", use_safetensors=True, variant='fp16').to(torch.float16).to(device)
     return pipe, device
 
 def parse_sub_prompts(text):