Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import numpy as np
 import torch
+import os
 
 with open("knowledge.txt", "r", encoding="utf-8") as file:
   knowledge_base = file.read()
@@ -70,8 +71,7 @@ top_results = get_top_chunks("Your account has been compromised", chunk_embeddin
 
 cleaned_chunks = preprocess_text(knowledge_base) 
 
-client = InferenceClient("Qwen/Qwen2.5-7B-Instruct")
-
+InferenceClient("Qwen/Qwen2.5-7B-Instruct", token=os.getenv("BYTESHIELD_TOKEN"))
 def respond(message, history):
     top_chunks = get_top_chunks(message, chunk_embeddings, cleaned_chunks)
     context = "\n".join(top_chunks)