Update models.py

models.py

@@ -1,22 +1,24 @@
-# models.py
-
 import gradio as gr
-import spaces
 import types
 
 # Create a local "transformers_gradio" object to mimic the missing package
 transformers_gradio = types.SimpleNamespace(registry="huggingface")
 
-# 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)
+# Load Gradio interfaces with the appropriate registry
+demo_distill = 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)
+
+# Example usage: Assign one of the models as the default demo
+demo = demo_distill
 
-# Example of using spaces (assuming `spaces.GPU()` is valid in your environment)
-demo.fn = spaces.GPU()(demo.fn)
+# Add functionality to demo
+if hasattr(demo, "fn"):
+    demo.fn = demo.fn
 
-for fn in demo.fns.values():
-    fn.api_name = False
+if hasattr(demo, "fns"):
+    for fn in demo.fns.values():
+        fn.api_name = False
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()