fix: set env via hf ui
Browse files
app.py
CHANGED
@@ -1,4 +1,3 @@
|
|
1 |
-
import os
|
2 |
from typing import Any, Dict
|
3 |
|
4 |
import cv2
|
@@ -13,7 +12,6 @@ from src.plot_utils import export_mask
|
|
13 |
|
14 |
from spaces import GPU
|
15 |
|
16 |
-
os.environ["ZEROGPU_V2"] = "true"
|
17 |
|
18 |
@GPU()
|
19 |
def predict(model_choice, annotations: Dict[str, Any]):
|
|
|
|
|
1 |
from typing import Any, Dict
|
2 |
|
3 |
import cv2
|
|
|
12 |
|
13 |
from spaces import GPU
|
14 |
|
|
|
15 |
|
16 |
@GPU()
|
17 |
def predict(model_choice, annotations: Dict[str, Any]):
|