jost commited on
Commit
199404a
·
verified ·
1 Parent(s): 2913e78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -88,18 +88,18 @@ def predict(
88
 
89
  client = OpenAI(base_url=togetherai_base_url, api_key=togetherai_api_key)
90
 
91
- response1 = client.completions.create(
92
  model=model1,
93
  prompt=prompt,
94
- temperature=0.7,
95
- max_tokens=1000).choices[0].text
96
-
97
- response2 = client.completions.create(
98
  model=model2,
99
  prompt=prompt,
100
- temperature=0.7,
101
- max_tokens=1000).choices[0].text
102
-
103
  return response1, response2
104
 
105
  def update_political_statement_options(test_type):
 
88
 
89
  client = OpenAI(base_url=togetherai_base_url, api_key=togetherai_api_key)
90
 
91
+ response1 = client.chat.completions.create(
92
  model=model1,
93
  prompt=prompt,
94
+ temperature=temperature,
95
+ max_tokens=1000).choices[0].message.content
96
+
97
+ response2 = client.chat.completions.create(
98
  model=model2,
99
  prompt=prompt,
100
+ temperature=temperature,
101
+ max_tokens=1000).choices[0].message.content
102
+
103
  return response1, response2
104
 
105
  def update_political_statement_options(test_type):