Gradio-LLM-demo / app.py
MakiAi's picture
Update app.py
4b1d975 verified
raw
history blame contribute delete
213 Bytes
import gradio as gr
import spaces
import transformers_gradio
demo = gr.load(name="AIDC-AI/Marco-o1", src=transformers_gradio.registry)
demo.fn = spaces.GPU()(demo.fn)
if __name__ == "__main__":
demo.launch()