Tj commited on
Commit
c6a2b84
·
1 Parent(s): 412460e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -11
app.py CHANGED
@@ -98,18 +98,19 @@ def load_recommender():
98
  chunks = text_to_chunks(texts, start_page=1)
99
  recommender.fit(chunks)
100
  return 'Corpus Loaded.'
 
101
  def generate_text(prompt, engine="text-davinci-003"):
102
- openai.api_key = OPENAI_API_KEY
103
- completions = openai.Completion.create(
104
- engine=engine,
105
- prompt=prompt,
106
- max_tokens=512,
107
- n=1,
108
- stop=None,
109
- temperature=0.7,
110
- )
111
- message = completions.choices[0].text
112
- return message
113
 
114
  def generate_answer(question):
115
  topn_chunks = recommender(question)
 
98
  chunks = text_to_chunks(texts, start_page=1)
99
  recommender.fit(chunks)
100
  return 'Corpus Loaded.'
101
+
102
  def generate_text(prompt, engine="text-davinci-003"):
103
+ openai.api_key = OPENAI_API_KEY
104
+ completions = openai.Completion.create(
105
+ engine=engine,
106
+ prompt=prompt,
107
+ max_tokens=512,
108
+ n=1,
109
+ stop=None,
110
+ temperature=0.7,
111
+ )
112
+ message = completions.choices[0].text
113
+ return message
114
 
115
  def generate_answer(question):
116
  topn_chunks = recommender(question)