Spaces:
Running
on
Zero
Running
on
Zero
ohayonguy
commited on
Commit
•
0d4c368
1
Parent(s):
e0edc6d
trying to fix cuda init
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import os
|
2 |
if os.getenv('SPACES_ZERO_GPU') == "true":
|
3 |
os.environ['SPACES_ZERO_GPU'] = "1"
|
|
|
4 |
import cv2
|
5 |
import gradio as gr
|
6 |
import torch
|
@@ -8,7 +9,6 @@ from basicsr.archs.srvgg_arch import SRVGGNetCompact
|
|
8 |
from basicsr.utils import img2tensor, tensor2img
|
9 |
from facexlib.utils.face_restoration_helper import FaceRestoreHelper
|
10 |
from realesrgan.utils import RealESRGANer
|
11 |
-
import spaces
|
12 |
|
13 |
from lightning_models.mmse_rectified_flow import MMSERectifiedFlow
|
14 |
|
|
|
1 |
import os
|
2 |
if os.getenv('SPACES_ZERO_GPU') == "true":
|
3 |
os.environ['SPACES_ZERO_GPU'] = "1"
|
4 |
+
import spaces
|
5 |
import cv2
|
6 |
import gradio as gr
|
7 |
import torch
|
|
|
9 |
from basicsr.utils import img2tensor, tensor2img
|
10 |
from facexlib.utils.face_restoration_helper import FaceRestoreHelper
|
11 |
from realesrgan.utils import RealESRGANer
|
|
|
12 |
|
13 |
from lightning_models.mmse_rectified_flow import MMSERectifiedFlow
|
14 |
|