Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
107 |
-
|
108 |
model=engine,
|
109 |
-
messages=[
|
110 |
-
|
|
|
|
|
111 |
max_tokens=max_tokens,
|
112 |
-
n=1,
|
113 |
temperature=temperature
|
114 |
-
|
115 |
-
|
116 |
-
|
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)
|