import gradio as gr
import spaces
import transformers_gradio

demo = gr.load(name="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", src=transformers_gradio.registry)
demo = gr.load(name="deepseek-ai/DeepSeek-R1", src=transformers_gradio.registry)
demo = gr.load(name="deepseek-ai/DeepSeek-R1-Zero", src=transformers_gradio.registry)


demo.fn = spaces.GPU()(demo.fn)

for fn in demo.fns.values():
    fn.api_name = False

if __name__ == "__main__":
    demo.launch()