Spaces:

prithivMLmods
/

CUA-GUI-Operator

Running on Zero

prithivMLmods commited on 10 days ago

Commit

f8804bb

verified ·

1 Parent(s): 73b010f

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -126,7 +126,7 @@ model_v = Qwen2_5_VLForConditionalGeneration.from_pretrained(
 # Load UI-TARS-1.5-7B
 print("Loading UI-TARS-1.5-7B...")
 # Note: Using the official SFT repo. Adjust if you have a specific private repo.
-MODEL_ID_X = "bytedance/UI-TARS-7B"
 processor_x = AutoProcessor.from_pretrained(MODEL_ID_X, trust_remote_code=True)
 model_x = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_X,

 # Load UI-TARS-1.5-7B
 print("Loading UI-TARS-1.5-7B...")
 # Note: Using the official SFT repo. Adjust if you have a specific private repo.
+MODEL_ID_X = "ByteDance-Seed/UI-TARS-1.5-7B"
 processor_x = AutoProcessor.from_pretrained(MODEL_ID_X, trust_remote_code=True)
 model_x = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_X,