Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -44,13 +44,13 @@ processor = AutoProcessor.from_pretrained(
|
|
44 |
|
45 |
try:
|
46 |
model = AutoModelForCausalLM.from_pretrained(
|
47 |
-
"
|
48 |
torch_dtype=torch.float16,
|
49 |
trust_remote_code=True,
|
50 |
).to("cuda")
|
51 |
except:
|
52 |
model = AutoModelForCausalLM.from_pretrained(
|
53 |
-
"
|
54 |
torch_dtype=torch.float16,
|
55 |
trust_remote_code=True,
|
56 |
)
|
|
|
44 |
|
45 |
try:
|
46 |
model = AutoModelForCausalLM.from_pretrained(
|
47 |
+
"banao-tech/OmniParse",
|
48 |
torch_dtype=torch.float16,
|
49 |
trust_remote_code=True,
|
50 |
).to("cuda")
|
51 |
except:
|
52 |
model = AutoModelForCausalLM.from_pretrained(
|
53 |
+
"banao-tech/OmniParse",
|
54 |
torch_dtype=torch.float16,
|
55 |
trust_remote_code=True,
|
56 |
)
|