Update app.py

app.py

@@ -29,7 +29,7 @@ def search(query):
     with requests.Session() as session:
         resp = session.get(
             url="https://www.google.com/search",
-            headers={"User-Agent": "Mozilla/5.0"},
+            headers={"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/111.0"},
             params={"q": term, "num": 3, "udm": 14},
             timeout=5,
             verify=None,
@@ -41,7 +41,7 @@ def search(query):
             link = result.find("a", href=True)
             link = link["href"]
             try:
-                webpage = session.get(link, headers={"User-Agent": "Mozilla/5.0"}, timeout=5, verify=False)
+                webpage = session.get(link, headers={"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/111.0"}, timeout=5, verify=False)
                 webpage.raise_for_status()
                 visible_text = extract_text_from_webpage(webpage.text)
                 if len(visible_text) > max_chars_per_page:
@@ -52,7 +52,7 @@ def search(query):
     return all_results
 
 # Initialize inference clients for different models
-client = InferenceClient("google/gemma-1.1-7b-it")
+client_gemma = InferenceClient("google/gemma-1.1-7b-it")
 client_mixtral = InferenceClient("NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO")
 client_llama = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct")
 
@@ -87,7 +87,7 @@ def respond(message, history):
     messages.append({"role": "user", "content": f'[SYSTEM]You are a helpful assistant. You have access to the following functions: \n {str(functions_metadata)}\n\nTo use these functions respond with:\n<functioncall> {{ "name": "function_name", "arguments": {{ "arg_1": "value_1", "arg_1": "value_1", ... }} }}  </functioncall>  [USER] {message} {vqa}'})
 
     # Call the LLM for response generation
-    response = client.chat_completion(messages, max_tokens=150)
+    response = client_gemma.chat_completion(messages, max_tokens=150)
     response = str(response)
     try:
         response = response[int(response.find("{")):int(response.index("</"))]