Update app.py

app.py

@@ -1,22 +1,25 @@
 import json
 import os
 from flask import Flask, request, jsonify
-from transformers import pipeline
-
-
 from transformers import pipeline
 from huggingface_hub import login
 
-# Log in with your Hugging Face token
-login(token="your_huggingface_token")
-
-# Load the model
-llm = pipeline("text-generation", model="tiiuae/falcon-1b-instruct")
+# Retrieve Hugging Face token securely
+hf_token = os.getenv("chatbot")
 
+if hf_token is None:
+    raise ValueError("Hugging Face token is missing! Set HF_TOKEN in your environment variables or Secrets.")
 
+# Log in with Hugging Face token
+login(token=hf_token)
 
+# Load the model (Falcon-1B)
+try:
+    llm = pipeline("text-generation", model="tiiuae/falcon-1b-instruct", token=hf_token)
+except Exception as e:
+    raise RuntimeError(f"Error loading model: {str(e)}")
 
-# Load the schemes data from JSON if available
+# Load the schemes data from JSON file
 SCHEMES_FILE = "schemes.json"
 
 if os.path.exists(SCHEMES_FILE):
@@ -32,7 +35,7 @@ def find_scheme_info(query):
     for scheme in schemes_data.get("schemes", []):
         if query.lower() in scheme["name"].lower() or query.lower() in scheme["description"].lower():
             return scheme["description"]
-    return None  # Return None instead of a default message
+    return None
 
 # Function to generate chatbot response
 def chatbot_response(query):