Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,25 +12,6 @@ except ImportError:
|
|
| 12 |
def GPU(duration=120):
|
| 13 |
return lambda fn: fn
|
| 14 |
|
| 15 |
-
# โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ
|
| 16 |
-
# โก ้่ฟ openmim ๅฎ่ฃ
mmcv / mmsegmentation
|
| 17 |
-
# ๏ผPyPI ๆ Python 3.12 ้ข็ผ่ฏ wheel๏ผๅฟ
้กป่ตฐ OpenMMLab CDN๏ผ
|
| 18 |
-
# โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ
|
| 19 |
-
import subprocess, sys
|
| 20 |
-
|
| 21 |
-
def _mim_install(pkg: str):
|
| 22 |
-
"""ไฝฟ็จ mim ๅฎ่ฃ
๏ผ่ชๅจๅน้
ๅฝๅ torch/CUDA ็ๆฌ็้ข็ผ่ฏ wheelใ"""
|
| 23 |
-
subprocess.check_call(
|
| 24 |
-
[sys.executable, "-m", "mim", "install", "-q", pkg]
|
| 25 |
-
)
|
| 26 |
-
|
| 27 |
-
try:
|
| 28 |
-
import mmcv # ๅทฒๅฎ่ฃ
ๅ่ทณ่ฟ
|
| 29 |
-
except ImportError:
|
| 30 |
-
print("๐ฆ Installing mmcv via openmim ...")
|
| 31 |
-
_mim_install("mmcv>=2.1.0")
|
| 32 |
-
print("โ
mmcv installed.")
|
| 33 |
-
|
| 34 |
import os, io
|
| 35 |
import torch
|
| 36 |
import numpy as np
|
|
|
|
| 12 |
def GPU(duration=120):
|
| 13 |
return lambda fn: fn
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
import os, io
|
| 16 |
import torch
|
| 17 |
import numpy as np
|