Spaces:
Runtime error
Runtime error
BusinessDev
commited on
Commit
•
7cb60ed
1
Parent(s):
7e207f6
First
Browse files
app.py
ADDED
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from transformers import MBartForConditionalGeneration, MBart50Tokenizer
|
2 |
+
|
3 |
+
# Load the model and tokenizer
|
4 |
+
model_name = "LocalDoc/mbart_large_qa_azerbaijan"
|
5 |
+
tokenizer = MBart50Tokenizer.from_pretrained(model_name, src_lang="en_XX", tgt_lang="az_AZ")
|
6 |
+
model = MBartForConditionalGeneration.from_pretrained(model_name)
|
7 |
+
|
8 |
+
def answer_question(context, question):
|
9 |
+
# Prepare input text
|
10 |
+
input_text = f"context: {context} question: {question}"
|
11 |
+
inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True, padding="max_length")
|
12 |
+
|
13 |
+
# Generate answer
|
14 |
+
outputs = model.generate(
|
15 |
+
input_ids=inputs["input_ids"],
|
16 |
+
attention_mask=inputs["attention_mask"],
|
17 |
+
max_length=128,
|
18 |
+
num_beams=5,
|
19 |
+
early_stopping=True
|
20 |
+
)
|
21 |
+
|
22 |
+
# Decode the answer
|
23 |
+
answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
24 |
+
return answer
|
25 |
+
|
26 |
+
# Example usage
|
27 |
+
context = """Ey iman gətirənlər! Səbir edərək və namaz qılaraq Allahdan kömək istəyin! Şübhəsiz ki, Allah səbir edənlərlədir.
|
28 |
+
Ey iman gətirənlər, əgər yalnız Allaha ibadət edirsinizsə, sizə verdiyimiz ruzilərin təmiz olanlarından yeyin və Ona şükür edin!
|
29 |
+
Ey iman gətirənlər! Oruc tutmaq sizdən əvvəlkilərə fərz buyurulduğu kimi sizə də fərz buyuruldu ki, bəlkə, günahlardan çəkinəsiniz!
|
30 |
+
Ey iman gətirənlər, bütövlüklə İslama daxil olun! Şeytanın addımlarını izləməyin! Şübhəsiz ki, o sizin açıq-aydın düşməninizdir.
|
31 |
+
Ey iman gətirənlər! Həqiqətən, içki, qumar, bütlər və fal oxları şeytan əməli olan murdar şeylərdir. Bunlardan uzaq durun ki, bəlkə nicat tapasınız!
|
32 |
+
Ey iman gətirənlər! “Allah yolunda döyüşə çıxın!” - deyildiyi zaman sizə nə oldu ki, yerə yapışıb qaldınız. Axirətin əvəzinə dünya həyatına razı oldunuz?
|
33 |
+
Dünya malı axirət nemətlərinin yanında çox az bir şeydir.
|
34 |
+
Həqiqətən, insan çox zalım və çox nankordur."""
|
35 |
+
|
36 |
+
question = "Allah kiminlədir ?"
|
37 |
+
|
38 |
+
answer = answer_question(context, question)
|
39 |
+
print(answer)
|