rodrigomasini commited on
Commit
4adf8b1
·
verified ·
1 Parent(s): ee40bdf

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -6
main.py CHANGED
@@ -1,12 +1,8 @@
1
  import gradio as gr
2
  import helper
3
- import spaces
4
 
5
- app=gr.load(
6
- name="AIDC-AI/Marco-o1",
7
- src=helper.registry,
8
- )
9
- app.fn = spaces.GPU()(app.fn)
10
 
11
  if __name__ == "__main__":
12
  app.launch()
 
1
  import gradio as gr
2
  import helper
 
3
 
4
+ # Create the Gradio interface using helper.registry
5
+ app = helper.registry(name="marco-o1")
 
 
 
6
 
7
  if __name__ == "__main__":
8
  app.launch()