Update app.py

app.py

@@ -6,6 +6,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cpu"
 
 access_token = os.getenv("access_token")
+privateurl = os.getenv("privateurl")
 
 tokenizer1 = AutoTokenizer.from_pretrained("Qwen/Qwen2-0.5B-Instruct")
 tokenizer2 = AutoTokenizer.from_pretrained("google/gemma-2-2b-it", token=access_token)
@@ -35,7 +36,7 @@ app = FastAPI()
 async def read_root():
     return {"Hello": "World!"}
 
-def modelResp1(id, prompt):
+def modelResp1(cookie, id, token, prompt):
     messages = [
         {"role": "system", "content": "You are a helpful assistant, Sia, developed by Sushma. You will response in polity and brief."},
         {"role": "user", "content": "Who are you?"},
@@ -53,10 +54,11 @@ def modelResp1(id, prompt):
       echo=False,  # Whether to echo the prompt
     )
     response = output['choices'][0]['text']
-
-    privatepayload['target'] = f"{id}"
-    privatepayload['content'] = response
-    requests.post(privateurl, headers=headers, data=privatepayload)
+    headers['Cookie'] = f"{cookie}"
+    payload['token'] = f"{token}"
+    payload['target'] = f"{id}"
+    payload['content'] = response
+    requests.post(privateurl, headers=headers, data=payload)
 
 def modelResp2(prompt):
     messages = [
@@ -103,7 +105,7 @@ def modelResp3(prompt):
 async def modelApi(data: dict):
     id = data.get("target_id")
     prompt = data.get("prompt")
-    modelResp1(id, prompt)
+    modelResp1(cookie, id, token, prompt)
     
 @app.post("/modelapi2")
 async def modelApi(data: dict):
@@ -123,7 +125,7 @@ headers = {
     'Accept-Encoding': 'gzip, deflate, br',
     'Accept-Language': 'en-US,en;q=0.9',
     'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
-    'Cookie': 'PHPSESSID=g0r2los456eaalf7epddmehabi; bc_userid=52113; bc_utk=aef8ec9314d8b64477cd86c80591f7e0c90f3faa',
+    'Cookie': '',
     'Sec-Ch-Ua': '"Opera";v="95", "Chromium";v="109", "Not;A=Brand";v="24"',
     'Sec-Ch-Ua-Mobile': '?0',
     'Sec-Ch-Ua-Platform': '"Windows"',