2022-12-26 15:43:10 +00:00
|
|
|
from typing import Any
|
|
|
|
|
|
2022-12-25 17:15:24 +00:00
|
|
|
import sys
|
|
|
|
|
from pathlib import Path
|
|
|
|
|
|
|
|
|
|
from transformers import pipeline
|
|
|
|
|
|
|
|
|
|
sys.path.append(str(Path(__file__).parent.parent.parent) + "/tools/NLP/data")
|
|
|
|
|
import internet
|
|
|
|
|
|
2022-12-26 15:43:10 +00:00
|
|
|
QA_MODEL = pipeline("question-answering")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def answer(query: str) -> Any:
|
|
|
|
|
global QA_MODEL
|
|
|
|
|
results = internet.google(query)
|
|
|
|
|
return (QA_MODEL(question=query, context=str(results[0])), results[1])
|
