BusinessDev commited on
Commit
7cb60ed
1 Parent(s): 7e207f6
Files changed (1) hide show
  1. app.py +39 -0
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)