cogcorp commited on
Commit
37c10f7
·
1 Parent(s): a4b14b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -103,17 +103,18 @@ def load_recommender(paths, start_page=1):
103
  recommender.fit(chunks)
104
  return 'Corpus Loaded.'
105
 
106
- def generate_text(prompt, engine='gpt-3.5-turbo', max_tokens=2050, temperature=0.8):
107
- response = openai.ChatCompletion.create(
108
  model=engine,
109
- messages=[{"role": "system", "content": "You are a research assistant"},
110
- {"role": "user", "content": prompt}],
 
 
111
  max_tokens=max_tokens,
112
- n=1,
113
  temperature=temperature
114
- )
115
- return response.choices[0].message['content']
116
- return None
117
 
118
  def generate_answer(question):
119
  topn_chunks = recommender(question)
 
103
  recommender.fit(chunks)
104
  return 'Corpus Loaded.'
105
 
106
+ def generate_text(prompt, engine='gpt-3.5-turbo', max_tokens=2048, temperature=0.8):
107
+ response = openai.ChatCompletion.create(
108
  model=engine,
109
+ messages=[
110
+ {"role": "system", "content": "You are a research assistant"},
111
+ {"role": "user", "content": prompt}
112
+ ],
113
  max_tokens=max_tokens,
 
114
  temperature=temperature
115
+ )
116
+ return response.choices[0].message['content']
117
+
118
 
119
  def generate_answer(question):
120
  topn_chunks = recommender(question)