Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,26 +5,24 @@ import kagglehub
|
|
| 5 |
from threading import Thread
|
| 6 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
|
| 7 |
|
| 8 |
-
# 🚀
|
| 9 |
-
#
|
| 10 |
-
os.environ["
|
| 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 |
-
# استخدام المسار ال
|
| 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 |
-
|
| 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__":
|