fix omniparser server; fix yyolo dependency

2025-02-01 14:49:04 -08:00
parent 80f6be73e1
commit 3d981833e6
2 changed files with 2 additions and 2 deletions
--- a/computer_use_demo/omniparserserver/omniparserserver.py
+++ b/computer_use_demo/omniparserserver/omniparserserver.py
@@ -48,4 +48,4 @@ async def root():
    return {"message": "Omniparser API ready"}

 if __name__ == "__main__":
-    uvicorn.run("remote_request:app", host=args.host, port=args.port, reload=True)
+    uvicorn.run("omniparserserver:app", host=args.host, port=args.port, reload=True)
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,7 +4,7 @@ torchvision
 supervision==0.18.0
 openai==1.3.5
 transformers
-ultralytics==8.1.24
+ultralytics==8.3.70
 azure-identity
 numpy
 opencv-python