Spaces:
Running on Zero
Running on Zero
Update default model to nvidia/LocateAnything-3B and transformers version
Browse files- app.py +1 -1
- 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", "
|
| 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.
|
| 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
|