Update app.py

app.py

@@ -10,6 +10,23 @@ client = InferenceClient(model=HF_MODEL, token=HF_TOKEN)
 # Persistent bot knowledge state
 bot_knowledge = {"dataset": None}
 
+def load_model(hf_token):
+    """Loads the model and tokenizer using the provided Hugging Face token."""
+    global model, tokenizer
+
+    # Set the token for Hugging Face authentication
+    HfFolder.save_token(hf_token, 'huggingface')
+
+    # Load the model
+    try:
+        with gr.Progress() as progress:
+            progress(0, "Loading model...")
+            model = AutoModelForCausalLM.from_pretrained(DEFAULT_MODEL)
+            tokenizer = AutoTokenizer.from_pretrained(DEFAULT_MODEL)
+            progress(100, "Model loaded successfully!")
+        return "Model loaded successfully!"
+    except Exception as e:
+        return f"Error loading model: {str(e)}"
 
 # Train chatbot by setting the dataset
 def train_chatbot(dataset):