Update app.py

app.py

@@ -19,12 +19,12 @@ class OmniParser:
         self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         # Initialize processor and model
         self.processor = AutoProcessor.from_pretrained(
-            "microsoft/OmniParser",
+            "microsoft/Florence-2-base",
             trust_remote_code=True,
             cache_dir="/code/.cache"
         )
         self.model = AutoModelForVisualQuestionAnswering.from_pretrained(
-            "microsoft/OmniParser",
+            "microsoft/OmniParser/icon_caption_florence",
             trust_remote_code=True,
             cache_dir="/code/.cache"
         ).to(self.device)