omarkashif commited on
Commit
b832b0a
·
verified ·
1 Parent(s): cd8077b

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +6 -6
src/streamlit_app.py CHANGED
@@ -36,10 +36,10 @@ def get_rewritten_query(user_query):
36
  ]
37
  try:
38
  resp = client.chat.completions.create(
39
- model="gpt-5-nano",
40
  messages=messages,
41
- # temperature=0.1,
42
- # max_completion_tokens=400
43
  )
44
  rewritten = resp.choices[0].message.content.strip()
45
  except Exception as e:
@@ -116,10 +116,10 @@ def generate_response(user_query, docs):
116
  f"Question:\n{user_query}"})
117
  try:
118
  resp = client.chat.completions.create(
119
- model="gpt-5-nano",
120
  messages=messages,
121
- # temperature=0.1,
122
- # max_completion_tokens=900
123
  )
124
  reply = resp.choices[0].message.content.strip()
125
  except Exception as e:
 
36
  ]
37
  try:
38
  resp = client.chat.completions.create(
39
+ model="gpt-4.1-mini",
40
  messages=messages,
41
+ temperature=0.1,
42
+ max_tokens=400
43
  )
44
  rewritten = resp.choices[0].message.content.strip()
45
  except Exception as e:
 
116
  f"Question:\n{user_query}"})
117
  try:
118
  resp = client.chat.completions.create(
119
+ model="gpt-4.1-mini",
120
  messages=messages,
121
+ temperature=0.1,
122
+ max_tokens=900
123
  )
124
  reply = resp.choices[0].message.content.strip()
125
  except Exception as e: