ONYX-APP commited on
Commit
1fa09da
·
verified ·
1 Parent(s): 71c6a10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -13
app.py CHANGED
@@ -5,26 +5,24 @@ import kagglehub
5
  from threading import Thread
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
7
 
8
- # 🚀 إعداد توكن Kaggle
9
- # ملاحظة: جربي استخدام التوكن كاملاً مع KGAT_
10
- os.environ["KAGGLE_USERNAME"] = "ruicompany"
11
- os.environ["KAGGLE_KEY"] = "KGAT_e80cdae2ef667fb69fff8585611b878f"
12
 
13
  # 🚀 تسريع CPU
14
  torch.set_num_threads(os.cpu_count())
15
 
16
  print("--- 📥 Downloading Model from Kaggle ---")
17
  try:
18
- # استخدام المسار الكامل مع تحديد الإصدار 1
19
- model_path = kagglehub.model_download("ruicompany/onyx-gpt2large/transformers/1gb-stable/1")
20
- except Exception as e:
21
- print(f"Error during download: {e}")
22
- # محاولة بديلة بدون رقم الإصدار في حال فشل الأول
23
  model_path = kagglehub.model_download("ruicompany/onyx-gpt2large/transformers/1gb-stable")
 
 
 
 
 
24
 
25
- print(f"--- 💎 Initializing ONYX from: {model_path} ---")
26
-
27
- # ... بقية الكود كما هو ...
28
  tokenizer = AutoTokenizer.from_pretrained(model_path)
29
  if tokenizer.pad_token is None:
30
  tokenizer.pad_token = tokenizer.eos_token
@@ -69,9 +67,10 @@ def chat_onyx(message, history):
69
  partial += new_text
70
  yield partial
71
 
 
72
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
73
  gr.Markdown("# 💎 ONYX AI - GPT2 Large Mode")
74
- gr.Markdown("مرحباً مهندسة روان، المودل قيد التحميل والتشغيل.")
75
  gr.ChatInterface(fn=chat_onyx)
76
 
77
  if __name__ == "__main__":
 
5
  from threading import Thread
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
7
 
8
+ # 🚀 الحل الجذري حسب تعليمات كاجل الجديدة
9
+ # نستخدم اسم المتغير KAGGLE_API_TOKEN كما طلبت الرسالة تماماً
10
+ os.environ["KAGGLE_API_TOKEN"] = "KGAT_e7d58c8edc599a3341c98d2f1295795f"
 
11
 
12
  # 🚀 تسريع CPU
13
  torch.set_num_threads(os.cpu_count())
14
 
15
  print("--- 📥 Downloading Model from Kaggle ---")
16
  try:
17
+ # استخدام المسار الصحيح للنسخة المستقرة
 
 
 
 
18
  model_path = kagglehub.model_download("ruicompany/onyx-gpt2large/transformers/1gb-stable")
19
+ print(f"--- ✅ Success! Model Path: {model_path} ---")
20
+ except Exception as e:
21
+ print(f"--- ❌ Download Failed: {e} ---")
22
+ # محاولة أخيرة بالمسار الافتراضي في حال وجود مشكلة في تسمية النسخة
23
+ model_path = kagglehub.model_download("ruicompany/onyx-gpt2large/transformers/default")
24
 
25
+ # تحميل الـ Tokenizer والـ Model من المسار المحلي
 
 
26
  tokenizer = AutoTokenizer.from_pretrained(model_path)
27
  if tokenizer.pad_token is None:
28
  tokenizer.pad_token = tokenizer.eos_token
 
67
  partial += new_text
68
  yield partial
69
 
70
+ # واجهة Gradio الاحترافية
71
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
72
  gr.Markdown("# 💎 ONYX AI - GPT2 Large Mode")
73
+ gr.Markdown(f"**Eng. Rawan**, ONYX is initializing via Kaggle API Token.")
74
  gr.ChatInterface(fn=chat_onyx)
75
 
76
  if __name__ == "__main__":