Update models.py

models.py

@@ -7,30 +7,16 @@ import types
 # 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)
 
-# If you still want to keep a default single "demo" reference
-# you can pick one. For example:
-demo = demo_qwen
+for fn in demo.fns.values():
+    fn.api_name = False
 
 if __name__ == "__main__":
-    # You could choose to launch one of them (or none, if controlling from app.py)
-    demo_qwen.launch()
+    demo.launch()