Update app.py

app.py

@@ -2,7 +2,7 @@ import re
 import shutil
 import subprocess
 import tempfile
-
+import json
 import gradio as gr
 import numpy as np
 import spaces
@@ -19,6 +19,7 @@ from transformers import (
     RfDetrForInstanceSegmentation,
     RfDetrForObjectDetection,
 )
+from pathlib import Path
 
 # -----------------------------
 # Config
@@ -76,6 +77,8 @@ VIDEO_LOW_CONF = 0.05
 
 _MODEL_CACHE: dict[str, tuple] = {}
 
+DETR_CONFIG_PATH = Path.cwd() / "config.json"
+DETR_CONFIG = json.loads(DETR_CONFIG_PATH.read_text(encoding="utf-8"))
 
 def is_seg_repo(repo: str) -> bool:
     return "-seg-" in repo
@@ -86,7 +89,7 @@ def load_model(repo: str):
     if repo not in _MODEL_CACHE:
         processor = AutoImageProcessor.from_pretrained(repo)
         cls = RfDetrForInstanceSegmentation if is_seg_repo(repo) else RfDetrForObjectDetection
-        model = cls.from_pretrained(repo).eval()
+        model = cls(**DETR_CONFIG).from_pretrained(repo).eval()
         _MODEL_CACHE[repo] = (processor, model)
     return _MODEL_CACHE[repo]