zero-gpu

app_marco_o1.py

@@ -1,16 +1,9 @@
-import os
-
+import gradio as gr
+import spaces
 import transformers_gradio
 
-from utils import get_app
-
-demo = get_app(
-    models=[
-        'AIDC-AI/Marco-o1'
-    ],
-    default_model="AIDC-AI/Marco-o1",
-    src=transformers_gradio.registry,
-)
+demo = gr.load(name="AIDC-AI/Marco-o1", src=transformers_gradio.registry)
+demo.fn = spaces.GPU()(demo.fn)
 
 if __name__ == "__main__":
     demo.launch()