Upload server.py with huggingface_hub
Browse files
server.py
CHANGED
|
@@ -70,7 +70,7 @@ def ask(body: Ask, request: Request):
|
|
| 70 |
from rag import answer
|
| 71 |
t = time.time()
|
| 72 |
res = answer(body.question.strip(), STATE["ret"], STATE["gen"], k=body.k, per_speech=2, lang=None if body.lang == "auto" else body.lang)
|
| 73 |
-
return {"question": res["question"], "answer": res["answer"], "model": res["model"], "retriever": res["retriever"], "lang": res["lang"],
|
| 74 |
"passages": [{"n": p["n"], "citation": p["citation"], "text": p["text"], "speech_id": p["speech_id"], "score": p["score"]} for p in res["passages"]],
|
| 75 |
"seconds": round(time.time() - t, 2)}
|
| 76 |
|
|
|
|
| 70 |
from rag import answer
|
| 71 |
t = time.time()
|
| 72 |
res = answer(body.question.strip(), STATE["ret"], STATE["gen"], k=body.k, per_speech=2, lang=None if body.lang == "auto" else body.lang)
|
| 73 |
+
return {"question": res["question"], "answer": res["answer"], "model": res["model"], "retriever": res["retriever"], "lang": res["lang"], "query_my": res.get("query_my"),
|
| 74 |
"passages": [{"n": p["n"], "citation": p["citation"], "text": p["text"], "speech_id": p["speech_id"], "score": p["score"]} for p in res["passages"]],
|
| 75 |
"seconds": round(time.time() - t, 2)}
|
| 76 |
|