SingleZombie
commited on
Commit
·
3592f41
1
Parent(s):
0a4892c
upd
Browse files- app.py +2 -0
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -4,6 +4,7 @@ import os
|
|
| 4 |
# uncomment the next line to use huggingface model in China
|
| 5 |
# os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
|
| 6 |
|
|
|
|
| 7 |
import cv2
|
| 8 |
import io
|
| 9 |
import gc
|
|
@@ -34,6 +35,7 @@ from model import build_model
|
|
| 34 |
from annotator.hed import HEDdetector
|
| 35 |
from annotator.canny import CannyDetector
|
| 36 |
# from annotator.midas import MidasDetector
|
|
|
|
| 37 |
|
| 38 |
import huggingface_hub
|
| 39 |
import shutil
|
|
|
|
| 4 |
# uncomment the next line to use huggingface model in China
|
| 5 |
# os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
|
| 6 |
|
| 7 |
+
import spaces
|
| 8 |
import cv2
|
| 9 |
import io
|
| 10 |
import gc
|
|
|
|
| 35 |
from annotator.hed import HEDdetector
|
| 36 |
from annotator.canny import CannyDetector
|
| 37 |
# from annotator.midas import MidasDetector
|
| 38 |
+
MidasDetector = None
|
| 39 |
|
| 40 |
import huggingface_hub
|
| 41 |
import shutil
|
requirements.txt
CHANGED
|
@@ -9,3 +9,4 @@ basicsr==1.4.2
|
|
| 9 |
numba==0.57.0
|
| 10 |
imageio-ffmpeg
|
| 11 |
gradio==4.27.0
|
|
|
|
|
|
| 9 |
numba==0.57.0
|
| 10 |
imageio-ffmpeg
|
| 11 |
gradio==4.27.0
|
| 12 |
+
spaces==0.26.1
|