Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
| 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.
|
| 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 |
|