Update main.py

main.py

@@ -50,7 +50,7 @@ try:
     ).to("cuda")
 except:
     model = AutoModelForCausalLM.from_pretrained(
-        "microsoft/OmniParser/icon_caption_florence",
+        "microsoft/OmniParser",
         torch_dtype=torch.float16,
         trust_remote_code=True,
     )