Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -103,9 +103,9 @@ def load_recommender(paths, start_page=1):
|
|
| 103 |
return 'Corpus Loaded.'
|
| 104 |
|
| 105 |
|
| 106 |
-
def generate_text(prompt):
|
| 107 |
completions = openai.Completion.create(
|
| 108 |
-
engine=
|
| 109 |
prompt=prompt,
|
| 110 |
max_tokens=512,
|
| 111 |
n=1,
|
|
@@ -117,6 +117,7 @@ def generate_text(prompt):
|
|
| 117 |
|
| 118 |
|
| 119 |
def generate_answer(question):
|
|
|
|
| 120 |
topn_chunks = recommender(question)
|
| 121 |
prompt = ""
|
| 122 |
prompt += 'search results:\n\n'
|
|
@@ -133,7 +134,7 @@ def generate_answer(question):
|
|
| 133 |
"answer should be short and concise.\n\nQuery: {question}\nAnswer: "
|
| 134 |
|
| 135 |
prompt += f"Query: {question}\nAnswer:"
|
| 136 |
-
answer = generate_text(prompt)
|
| 137 |
return answer
|
| 138 |
|
| 139 |
|
|
|
|
| 103 |
return 'Corpus Loaded.'
|
| 104 |
|
| 105 |
|
| 106 |
+
def generate_text(prompt,engine):
|
| 107 |
completions = openai.Completion.create(
|
| 108 |
+
engine=engine,
|
| 109 |
prompt=prompt,
|
| 110 |
max_tokens=512,
|
| 111 |
n=1,
|
|
|
|
| 117 |
|
| 118 |
|
| 119 |
def generate_answer(question):
|
| 120 |
+
engine = os.environ('Engine')
|
| 121 |
topn_chunks = recommender(question)
|
| 122 |
prompt = ""
|
| 123 |
prompt += 'search results:\n\n'
|
|
|
|
| 134 |
"answer should be short and concise.\n\nQuery: {question}\nAnswer: "
|
| 135 |
|
| 136 |
prompt += f"Query: {question}\nAnswer:"
|
| 137 |
+
answer = generate_text(prompt, engine)
|
| 138 |
return answer
|
| 139 |
|
| 140 |
|