bhaskartripathi commited on
Commit
fae8909
·
1 Parent(s): ea50b28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
app.py CHANGED
@@ -98,6 +98,19 @@ def load_recommender(path, start_page=1):
98
  recommender.fit(chunks)
99
  return 'Corpus Loaded.'
100
 
 
 
 
 
 
 
 
 
 
 
 
 
 
101
 
102
  def generate_answer(question,openAI_key):
103
  topn_chunks = recommender(question)
 
98
  recommender.fit(chunks)
99
  return 'Corpus Loaded.'
100
 
101
+ def generate_text(openAI_key,prompt, engine="text-davinci-003"):
102
+ openai.api_key = openAI_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
 
115
  def generate_answer(question,openAI_key):
116
  topn_chunks = recommender(question)