Update app.py

app.py

@@ -19,8 +19,8 @@ cookie_manager = stx.CookieManager()
 
 @st.cache_resource
 def get_vision_model():
-    model = AutoModelForVision2Seq.from_pretrained("ydshieh/kosmos-2-patch14-224", trust_remote_code=True)
-    processor = AutoProcessor.from_pretrained("ydshieh/kosmos-2-patch14-224", trust_remote_code=True)
+    model = AutoModelForVision2Seq.from_pretrained("microsoft/kosmos-2-patch14-224")
+    processor = AutoProcessor.from_pretrained("microsoft/kosmos-2-patch14-224")
     return model, processor
 
 # Function to get image caption via Kosmos2.