requesting ZeroGPU gpus

app.py

@@ -1,3 +1,4 @@
+import spaces
 import random
 from threading import Thread
 
@@ -331,5 +332,9 @@ with gr.Blocks() as demo:
     )
 
 
-if __name__ == "__main__":
+@spaces.GPU
+def main():
     demo.launch()
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -7,4 +7,5 @@ torchaudio==2.5.1
 torchvision==0.20.1
 tqdm==4.67.1
 transformers @ git+https://github.com/huggingface/transformers.git@1931a351408dbd1d0e2c4d6d7ee0eb5e8807d7bf
-trl @ git+https://github.com/sunildkumar/trl.git
+trl @ git+https://github.com/sunildkumar/trl.git
+spaces==0.32.0