Update main.py

main.py

@@ -1,8 +1,12 @@
 import gradio as gr
 import helper
+import spaces
 
-gr.load(
+app=gr.load(
     name="AIDC-AI/Marco-o1",
-    model_path='AIDC-AI/Marco-o1',
     src=helper.registry,
-).launch()
+)
+app.fn = spaces.GPU()(app.fn)
+
+if __name__ == "__main__":
+    app.launch()