akhaliq HF Staff commited on
Commit
81ae955
·
1 Parent(s): 2593ff5

Update default model to nvidia/LocateAnything-3B and transformers version

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -326,7 +326,7 @@ def generate_raw_prompt(task_type, category):
326
  # 模型初始化
327
  # ============================================================
328
  try:
329
- MODEL_PATH = os.environ.get("MODEL_PATH", "woshichaoren123/test001")
330
  GLOBAL_WORKER = EagleWorker(MODEL_PATH)
331
  except Exception as e:
332
  print(f"Failed to load model: {e}. Will run in Mock Mode.")
 
326
  # 模型初始化
327
  # ============================================================
328
  try:
329
+ MODEL_PATH = os.environ.get("MODEL_PATH", "nvidia/LocateAnything-3B")
330
  GLOBAL_WORKER = EagleWorker(MODEL_PATH)
331
  except Exception as e:
332
  print(f"Failed to load model: {e}. Will run in Mock Mode.")
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  opencv-python-headless==4.11.0.86
2
- transformers==4.51.0
3
  torch==2.8.0
4
  torchvision==0.23.0
5
  numpy==1.25.0
 
1
  opencv-python-headless==4.11.0.86
2
+ transformers==4.57.1
3
  torch==2.8.0
4
  torchvision==0.23.0
5
  numpy==1.25.0