Update app.py

app.py

@@ -1,27 +1,11 @@
 import gradio as gr
-from transformers import MT5ForConditionalGeneration, MT5Tokenizer
+from transformers import pipeline
 
-# Модельді жүктеу
-model = MT5ForConditionalGeneration.from_pretrained("model")
-tokenizer = MT5Tokenizer.from_pretrained("google/mt5-small")
+qa_pipeline = pipeline("question-answering", model="./model", tokenizer="./model")
 
-# Сұраққа жауап беру функциясы
-def answer_question(question, context):
-    input_text = f"Сұрақ: {question} Контекст: {context}"
-    input_ids = tokenizer(input_text, return_tensors="pt").input_ids
-    output_ids = model.generate(input_ids, max_length=128)
-    return tokenizer.decode(output_ids[0], skip_special_tokens=True)
+def answer_question(question):
+    context = open("data.txt", encoding="utf-8").read()
+    result = qa_pipeline(question=question, context=context)
+    return result["answer"]
 
-# Интерфейс
-iface = gr.Interface(
-    fn=answer_question,
-    inputs=[
-        gr.Textbox(label="Сұрақ жазыңыз"),
-        gr.Textbox(label="Контекст (мәтін үзіндісі)")
-    ],
-    outputs="text",
-    title="Қазақша AI-ассистент",
-    description="Сұрақ қойып, контекст бойынша жауап алыңыз (Мәліметтер қоры тақырыбы)"
-)
-
-iface.launch()
+gr.Interface(fn=answer_question, inputs="text", outputs="text", title="Қазақша Ассистент").launch()