Update app.py

app.py

@@ -1,14 +1,14 @@
 import gradio as gr
 import spaces
-import transformers_gradio
 from functools import lru_cache
 
 # Cache model loading to optimize performance
 @lru_cache(maxsize=3)
 def load_hf_model(model_name):
+    # Use the Gradio-built huggingface loader instead of transformers_gradio
     return gr.load(
         name=f"deepseek-ai/{model_name}",
-        src=transformers_gradio.registry,
+        src="huggingface",  # Changed from transformers_gradio.registry
         api_name="/chat"
     )
 
@@ -51,7 +51,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="DeepSeek Chatbot") as demo:
         """
         # DeepSeek Chatbot
         Created by [ruslanmv.com](https://ruslanmv.com/)
-
         This is a demo of different DeepSeek models. Select a model, type your message, and click "Submit".
         You can also adjust optional parameters like system message, max new tokens, temperature, and top-p.
         """
@@ -104,8 +103,8 @@ with gr.Blocks(theme=gr.themes.Soft(), title="DeepSeek Chatbot") as demo:
 
 # Add GPU support for Hugging Face Spaces
 demo.fn = spaces.GPU()(demo.fn)
-for fn in demo.fns.values(): 
+for fn in demo.fns.values():
     fn.api_name = False
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()