Spaces:
Running
Running
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()
|