Gyaneshere commited on
Commit
60ce791
·
verified ·
1 Parent(s): 0e4b9e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,4 +1,7 @@
1
  from transformers import launch_gradio_demo
2
  from tool import TextToImageTool
 
 
 
3
 
4
  launch_gradio_demo(TextToImageTool)
 
1
  from transformers import launch_gradio_demo
2
  from tool import TextToImageTool
3
+ import spaces
4
+
5
+ @spaces.GPU
6
 
7
  launch_gradio_demo(TextToImageTool)