Antoni Bigata commited on
Commit
892dd0f
·
1 Parent(s): 7746897

addapt for zerogpu

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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()