Subh775 commited on
Commit
7622e4a
·
verified ·
1 Parent(s): 2313bae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -376,7 +376,7 @@ from rfdetr import RFDETRSegPreview
376
  app = Flask(__name__, static_folder="static", static_url_path="/")
377
 
378
  # Checkpoint URL & local path
379
- CHECKPOINT_URL = "https://huggingface.co/Subh775/Segment-Tulsi-TFs-3/resolve/main/checkpoint_best_total.pth"
380
  CHECKPOINT_PATH = os.path.join("/tmp", "checkpoint_best_total.pth")
381
 
382
  MODEL_LOCK = threading.Lock()
@@ -554,7 +554,7 @@ def predict():
554
 
555
  # Parse input
556
  img: Optional[Image.Image] = None
557
- conf_threshold = 0.25
558
  show_labels = True
559
  show_confidence = True
560
 
 
376
  app = Flask(__name__, static_folder="static", static_url_path="/")
377
 
378
  # Checkpoint URL & local path
379
+ CHECKPOINT_URL = "https://huggingface.co/Subh775/Segment-Tulsi-TFs/resolve/main/checkpoint_best_total.pth"
380
  CHECKPOINT_PATH = os.path.join("/tmp", "checkpoint_best_total.pth")
381
 
382
  MODEL_LOCK = threading.Lock()
 
554
 
555
  # Parse input
556
  img: Optional[Image.Image] = None
557
+ conf_threshold = 0.05
558
  show_labels = True
559
  show_confidence = True
560