File size: 696 Bytes
aea6f00
cd9467b
7cc244b
cd9467b
 
 
 
 
7cc244b
eb02560
 
 
 
7cc244b
eb02560
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import gradio as gr
from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline

# Модель мен токенизаторды нақты локал жолмен жүктейміз
model = AutoModelForQuestionAnswering.from_pretrained("./model")
tokenizer = AutoTokenizer.from_pretrained("./model")

qa_pipeline = pipeline("question-answering", model=model, tokenizer=tokenizer)

def answer_question(question):
    context = open("data.txt", encoding="utf-8").read()
    result = qa_pipeline(question=question, context=context)
    return result["answer"]

gr.Interface(fn=answer_question, inputs="text", outputs="text", title="Қазақша Ассистент").launch()