Upload v2.py
Browse files
v2.py
CHANGED
@@ -20,18 +20,8 @@ from dartrs.utils import get_generation_config
|
|
20 |
import gradio as gr
|
21 |
from gradio.components import Component
|
22 |
|
23 |
-
try:
|
24 |
-
import spaces
|
25 |
-
except ImportError:
|
26 |
-
|
27 |
-
class spaces:
|
28 |
-
def GPU(*args, **kwargs):
|
29 |
-
return lambda x: x
|
30 |
-
|
31 |
-
|
32 |
from output import UpsamplingOutput
|
33 |
|
34 |
-
|
35 |
HF_TOKEN = os.getenv("HF_TOKEN", None)
|
36 |
|
37 |
V2_ALL_MODELS = {
|
|
|
20 |
import gradio as gr
|
21 |
from gradio.components import Component
|
22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
23 |
from output import UpsamplingOutput
|
24 |
|
|
|
25 |
HF_TOKEN = os.getenv("HF_TOKEN", None)
|
26 |
|
27 |
V2_ALL_MODELS = {
|