Update app.py

app.py

@@ -24,6 +24,7 @@ import time
 import timeout_decorator
 import socket
 import ipaddress
+from huggingface_hub import spaces
 
 # Setup logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -460,5 +461,8 @@ with gr.Blocks() as demo:
     )
 
 # Launch the Gradio App
-if __name__ == "__main__":
-    demo.launch()
+@spaces.GPU
+def launch_app():
+    demo.launch()
+
+launch_app()