Update app.py

app.py

@@ -1,57 +1,57 @@
-import os
-from flask import Flask, render_template, request, jsonify
-import requests
-import time
-from dotenv import load_dotenv
-
-load_dotenv()  # Load environment variables from .env file
-
-app = Flask(__name__)
-
-class APIInferenceChatbot:
-    def __init__(self, api_token):
-        self.api_url = "https://api-inference.huggingface.co/models/Ouiam123/Llama-2-7b-chat-finetune-tourism"
-        self.headers = {
-            "Authorization": f"Bearer {api_token}",
-            "Content-Type": "application/json"
-        }
-
-    def generate_response(self, input_text):
-        formatted_prompt = f"<s>[INST] {input_text} [/INST>"
-        payload = {
-            "inputs": formatted_prompt,
-            "parameters": {
-                "max_new_tokens": 500,
-                "temperature": 0.7,
-                "top_p": 0.95,
-                "repetition_penalty": 1.15
-            }
-        }
-        response = requests.post(self.api_url, headers=self.headers, json=payload, timeout=30)
-        if response.status_code == 503:
-            time.sleep(20)
-            response = requests.post(self.api_url, headers=self.headers, json=payload)
-        response.raise_for_status()
-        result = response.json()
-        return result[0].get('generated_text', '').strip() if isinstance(result, list) and result else str(result)
-
-# Get the API token from the environment variable
-api_token = os.getenv("HF_API_TOKEN")
-if not api_token:
-    raise ValueError("Hugging Face API token not found in environment variables!")
-
-# Initialize the chatbot with the API token from the environment
-chatbot = APIInferenceChatbot(api_token)
-
-@app.route('/')
-def home():
-    return render_template('index.html')  # Serve the HTML file
-
-@app.route('/chat', methods=['POST'])
-def chat():
-    message = request.json.get('message', '')
-    response = chatbot.generate_response(message)
-    return jsonify({'response': response})  # Return the response as JSON
-
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=True)
+import os
+from flask import Flask, render_template, request, jsonify
+import requests
+import time
+from dotenv import load_dotenv
+
+load_dotenv()  # Load environment variables from .env file
+
+app = Flask(__name__)
+
+class APIInferenceChatbot:
+    def __init__(self, api_token):
+        self.api_url = "https://api-inference.huggingface.co/models/Ouiam123/Llama-2-7b-chat-finetune-tourism"
+        self.headers = {
+            "Authorization": f"Bearer {api_token}",
+            "Content-Type": "application/json"
+        }
+
+    def generate_response(self, input_text):
+        formatted_prompt = f"<s>[INST] {input_text} [/INST>"
+        payload = {
+            "inputs": formatted_prompt,
+            "parameters": {
+                "max_new_tokens": 500,
+                "temperature": 0.7,
+                "top_p": 0.95,
+                "repetition_penalty": 1.15
+            }
+        }
+        response = requests.post(self.api_url, headers=self.headers, json=payload, timeout=30)
+        if response.status_code == 503:
+            time.sleep(20)
+            response = requests.post(self.api_url, headers=self.headers, json=payload)
+        response.raise_for_status()
+        result = response.json()
+        return result[0].get('generated_text', '').strip() if isinstance(result, list) and result else str(result)
+
+# Get the API token from the environment variable
+api_token = os.getenv("tok1")
+if not api_token:
+    raise ValueError("Hugging Face API token not found in environment variables!")
+
+# Initialize the chatbot with the API token from the environment
+chatbot = APIInferenceChatbot(api_token)
+
+@app.route('/')
+def home():
+    return render_template('index.html')  # Serve the HTML file
+
+@app.route('/chat', methods=['POST'])
+def chat():
+    message = request.json.get('message', '')
+    response = chatbot.generate_response(message)
+    return jsonify({'response': response})  # Return the response as JSON
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=True)