Update app.py

app.py

@@ -4,12 +4,12 @@ import torch
 from diffusers import Lumina2Pipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# Set up environment
-os.environ['CUDA_VISIBLE_DEVICES'] = "0"
+# # Set up environment
+# os.environ['CUDA_VISIBLE_DEVICES'] = "0"
 
 # Load models
 def load_models():
-    model_name = "/mnt/petrelfs/zhaoshitian/models/LeX-Enhancer-full"
+    model_name = "X-ART/LeX-Enhancer-full"
     
     model = AutoModelForCausalLM.from_pretrained(
         model_name,
@@ -19,7 +19,7 @@ def load_models():
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     
     pipe = Lumina2Pipeline.from_pretrained(
-        "/mnt/hwfile/alpha_vl/qilongwu/checkpoints/LeX-Lumina", 
+        "X-ART/LeX-Lumina", 
         torch_dtype=torch.bfloat16
     )
     pipe.to("cuda")