Update app.py
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ def rag_query(q: str) -> str:
|
|
| 36 |
)
|
| 37 |
context = "\n".join(results["documents"][0])
|
| 38 |
prompt = f"Context:\n{context}\n\nUser: {q}\nAssistant:"
|
| 39 |
-
out = model.
|
| 40 |
return out["choices"][0]["text"]
|
| 41 |
|
| 42 |
# 3. FastAPI app
|
|
|
|
| 36 |
)
|
| 37 |
context = "\n".join(results["documents"][0])
|
| 38 |
prompt = f"Context:\n{context}\n\nUser: {q}\nAssistant:"
|
| 39 |
+
out = model.create_completion(prompt=prompt, max_tokens=256, temperature=0.7)
|
| 40 |
return out["choices"][0]["text"]
|
| 41 |
|
| 42 |
# 3. FastAPI app
|