Update deepseek-ai.py

deepseek-ai.py

@@ -2,15 +2,24 @@ 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)
+# 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)
 
-demo.fn = spaces.GPU()(demo.fn)
-
-for fn in demo.fns.values():
+# 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__":
-    demo.launch()
+    demo1.launch()