Spaces:
Running
on
Zero
Running
on
Zero
Antoni Bigata
commited on
Commit
·
892dd0f
1
Parent(s):
7746897
addapt for zerogpu
Browse files
app.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
import tempfile
|
@@ -23,7 +24,6 @@ from inference_functions import (
|
|
23 |
)
|
24 |
from wordle_game import WordleGame
|
25 |
import torch.cuda.amp as amp # Import amp for mixed precision
|
26 |
-
import spaces
|
27 |
|
28 |
|
29 |
# Set default tensor type to float16 for faster computation
|
@@ -990,4 +990,6 @@ with gr.Blocks(title="Video Synchronization with Diffusion Models") as demo:
|
|
990 |
""")
|
991 |
|
992 |
if __name__ == "__main__":
|
|
|
|
|
993 |
demo.launch()
|
|
|
1 |
+
import spaces
|
2 |
import gradio as gr
|
3 |
import torch
|
4 |
import tempfile
|
|
|
24 |
)
|
25 |
from wordle_game import WordleGame
|
26 |
import torch.cuda.amp as amp # Import amp for mixed precision
|
|
|
27 |
|
28 |
|
29 |
# Set default tensor type to float16 for faster computation
|
|
|
990 |
""")
|
991 |
|
992 |
if __name__ == "__main__":
|
993 |
+
import spaces
|
994 |
+
|
995 |
demo.launch()
|