Spaces:
Build error
Build error
File size: 835 Bytes
7789ae0 116f473 eb7b2cf 116f473 7789ae0 116f473 7789ae0 116f473 7789ae0 116f473 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
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() |