Update app.py

app.py

@@ -1,13 +1,13 @@
 import gradio as gr
-import spaces
 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="huggingface",
+        src="huggingface",  # Changed from transformers_gradio.registry
         api_name="/chat"
     )
 
@@ -41,7 +41,8 @@ def chatbot(input_text, history, model_choice, system_message, max_new_tokens, t
     except Exception as e:
         assistant_response = f"Error: {str(e)}"
     
-    history.append((input_text, assistant_response))
+    history.append({"role": "user", "content": input_text})
+    history.append({"role": "assistant", "content": assistant_response})
     return history, history, ""
 
 # --- Gradio Interface ---
@@ -57,8 +58,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="DeepSeek Chatbot") as demo:
 
     with gr.Row():
         with gr.Column():
-            # Specify type='messages' to avoid deprecation warning
-            chatbot_output = gr.Chatbot(label="DeepSeek Chatbot", height=500, type="messages")
+            chatbot_output = gr.Chatbot(label="DeepSeek Chatbot", height=500, type='messages')
             msg = gr.Textbox(label="Your Message", placeholder="Type your message here...")
             with gr.Row():
                 submit_btn = gr.Button("Submit", variant="primary")
@@ -101,11 +101,5 @@ with gr.Blocks(theme=gr.themes.Soft(), title="DeepSeek Chatbot") as demo:
         [chatbot_output, chat_history, msg]
     )
 
-# Remove (or replace) the old `demo.fn = spaces.GPU()(demo.fn)` line
-# If you need GPU acceleration in Spaces, wrap demo.launch() with spaces.GPU()
-# as shown below. Otherwise, remove the spaces.GPU() call.
-
 if __name__ == "__main__":
-    # Launch the Gradio app with GPU from Hugging Face Spaces
-    # If you DON'T need GPU from `spaces`, just do: demo.launch()
-    spaces.GPU()(demo.launch)()
+    demo.launch()