Update models.py

models.py

@@ -2,21 +2,32 @@
 
 import gradio as gr
 import spaces
-import types
+import transformers_gradio
 
-# Create a local "transformers_gradio" object to mimic the missing package
-transformers_gradio = types.SimpleNamespace(registry="huggingface")
+# Load each model separately
+demo_qwen = gr.load(
+    name="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
+    src=transformers_gradio.registry
+)
+demo_r1 = gr.load(
+    name="deepseek-ai/DeepSeek-R1",
+    src=transformers_gradio.registry
+)
+demo_zero = gr.load(
+    name="deepseek-ai/DeepSeek-R1-Zero",
+    src=transformers_gradio.registry
+)
 
-# Now the gradio.load calls will work as if we had `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)
+# Ensure they run on GPU and disable api_name
+for demo in [demo_qwen, demo_r1, demo_zero]:
+    demo.fn = spaces.GPU()(demo.fn)
+    for fn in demo.fns.values():
+        fn.api_name = False
 
-# Example of using spaces (assuming `spaces.GPU()` is valid in your environment)
-demo.fn = spaces.GPU()(demo.fn)
-
-for fn in demo.fns.values():
-    fn.api_name = False
+# If you still want to keep a default single "demo" reference
+# you can pick one. For example:
+demo = demo_qwen
 
 if __name__ == "__main__":
-    demo.launch()
+    # You could choose to launch one of them (or none, if controlling from app.py)
+    demo_qwen.launch()