Update models.py

models.py

@@ -1,24 +1,5 @@
 import gradio as gr
-import types
 
-# Create a local "transformers_gradio" object to mimic the missing package
-transformers_gradio = types.SimpleNamespace(registry="huggingface")
-
-# 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
-
-# Add functionality to demo
-if hasattr(demo, "fn"):
-    demo.fn = demo.fn
-
-if hasattr(demo, "fns"):
-    for fn in demo.fns.values():
-        fn.api_name = False
-
-if __name__ == "__main__":
-    demo.launch()
+# Load and return the model interface
+def load_model():
+    return gr.Interface.load("models/deepseek-ai/DeepSeek-R1")