import gradio as gr
import spaces
import transformers_gradio

# Cargar los modelos de manera fija (sin contenido dinámico)
demo1 = gr.load(name="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", src=transformers_gradio.registry)
demo2 = gr.load(name="deepseek-ai/DeepSeek-R1", src=transformers_gradio.registry)
demo3 = gr.load(name="deepseek-ai/DeepSeek-R1-Zero", src=transformers_gradio.registry)

# Asignar GPU a las funciones de los modelos
demo1.fn = spaces.GPU()(demo1.fn)
demo2.fn = spaces.GPU()(demo2.fn)
demo3.fn = spaces.GPU()(demo3.fn)

# Desactivar la API para cada función si es necesario
for fn in demo1.fns.values():
    fn.api_name = False
for fn in demo2.fns.values():
    fn.api_name = False
for fn in demo3.fns.values():
    fn.api_name = False

# Lanzar la interfaz de Gradio
if __name__ == "__main__":
    demo1.launch()