Upload gradio_s3diff.py
Browse files- src/gradio_s3diff.py +1 -1
src/gradio_s3diff.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import gradio as gr
|
2 |
import os
|
3 |
import sys
|
@@ -17,7 +18,6 @@ from de_net import DEResNet
|
|
17 |
from s3diff_tile import S3Diff
|
18 |
from torchvision import transforms
|
19 |
from utils.wavelet_color import wavelet_color_fix, adain_color_fix
|
20 |
-
from spaces import GPU # 追加
|
21 |
|
22 |
tensor_transforms = transforms.Compose([
|
23 |
transforms.ToTensor(),
|
|
|
1 |
+
from spaces import GPU # 最初にインポート
|
2 |
import gradio as gr
|
3 |
import os
|
4 |
import sys
|
|
|
18 |
from s3diff_tile import S3Diff
|
19 |
from torchvision import transforms
|
20 |
from utils.wavelet_color import wavelet_color_fix, adain_color_fix
|
|
|
21 |
|
22 |
tensor_transforms = transforms.Compose([
|
23 |
transforms.ToTensor(),
|