Oranblock commited on
Commit
3962dd8
·
verified ·
1 Parent(s): 92199a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -10
app.py CHANGED
@@ -1,10 +1,10 @@
1
- import gradio as gr
2
- from transformers import GPT2LMHeadModel, GPT2Tokenizer
3
  import random
4
 
5
- # تحميل النموذج والتوكنيزر
6
- tokenizer = GPT2Tokenizer.from_pretrained("aubmindlab/aragpt2-medium")
7
- model = GPT2LMHeadModel.from_pretrained("aubmindlab/aragpt2-medium")
 
8
 
9
  # الحروف المقطعة
10
  harf_muqattaat = ['أ', 'ل', 'م', 'ص', 'ر', 'ك', 'ه', 'ي', 'ع', 'ط', 'س', 'ح', 'ق', 'ن']
@@ -29,8 +29,6 @@ def generate_words_from_muqattaat():
29
 
30
  return generated_text
31
 
32
- # إنشاء واجهة Gradio
33
- iface = gr.Interface(fn=generate_words_from_muqattaat, inputs=None, outputs="text", title="توليد كلمات عربية باستخدام الحروف المقطعة")
34
-
35
- # تشغيل التطبيق
36
- iface.launch()
 
1
+ from transformers import AutoTokenizer, AutoModelForCausalLM
 
2
  import random
3
 
4
+ # تحميل النموذج والتوكنيزر من Hugging Face
5
+ model_name = "EleutherAI/gpt-j-6B" # يمكنك أيضًا استخدام "aubmindlab/aragpt2-medium" لنموذج خاص بالعربية
6
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
7
+ model = AutoModelForCausalLM.from_pretrained(model_name)
8
 
9
  # الحروف المقطعة
10
  harf_muqattaat = ['أ', 'ل', 'م', 'ص', 'ر', 'ك', 'ه', 'ي', 'ع', 'ط', 'س', 'ح', 'ق', 'ن']
 
29
 
30
  return generated_text
31
 
32
+ # توليد الجمل
33
+ for _ in range(5):
34
+ print(generate_words_from_muqattaat())