Update app.py

app.py

@@ -53,8 +53,8 @@ SEARXNG_KEY = 'f9f07f93b37b8483aadb5ba717f556f3a4ac507b281b4ca01e6c6288aa3e3ae5'
 HF_TOKEN = os.getenv('HF_TOKEN')
 
 # Modify the client initialization to use the selected model
-def get_inference_client(model_name):
-    return InferenceClient(model_name, token=HF_TOKEN)
+
+client = InferenceClient(model_name, token=HF_TOKEN)
 
 # Initialize the similarity model
 similarity_model = SentenceTransformer('all-MiniLM-L6-v2')
@@ -555,7 +555,7 @@ def chat_function(message, history, num_results, scraper, max_chars, time_range,
     chat_history = "\n".join([f"{role}: {msg}" for role, msg in history])
     
     # Get the client for the selected model
-    client = get_inference_client(model_name)
+    client = client(model_name)
     
     response = search_and_scrape(
         query=message,