Spaces:

banao-tech
/

OmniPar

Sleeping

banao-tech commited on Dec 31, 2024

Commit

3d8571e

verified ·

1 Parent(s): 52027db

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -44,13 +44,13 @@ processor = AutoProcessor.from_pretrained(
 try:
     model = AutoModelForCausalLM.from_pretrained(
-        "microsoft/OmniParser",
         torch_dtype=torch.float16,
         trust_remote_code=True,
     ).to("cuda")
 except:
     model = AutoModelForCausalLM.from_pretrained(
-        "microsoft/OmniParser",
         torch_dtype=torch.float16,
         trust_remote_code=True,
     )

 try:
     model = AutoModelForCausalLM.from_pretrained(
+        "banao-tech/OmniParse",
         torch_dtype=torch.float16,
         trust_remote_code=True,
     ).to("cuda")
 except:
     model = AutoModelForCausalLM.from_pretrained(
+        "banao-tech/OmniParse",
         torch_dtype=torch.float16,
         trust_remote_code=True,
     )