Update app.py

app.py

@@ -1,42 +1,34 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
-
-model_name = "MahmoudIbrahim/Meta-LLama3-Instruct-Arabic"
-
-# ✅ نحمل النموذج بدون 4bit وبدون quantization (يشتغل على CPU)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(
-    model_name,
-    device_map="cpu",           # تشغيل على CPU
-    trust_remote_code=True,     # للسماح بتحميل الكود الخاص بالنموذج
-    torch_dtype="float32"       # نستخدم دقة عادية تناسب CPU
-)
+from transformers import pipeline
 
+# 🔹 استخدمنا النموذج الشعري بدلاً من النموذج العادي
 generator = pipeline(
     "text-generation",
-    model=model,
-    tokenizer=tokenizer
+    model="akhooli/gpt2-small-arabic-poetry"
 )
 
 def inspire(future, value, skill):
     if not future or not value or not skill:
-        return "⚠️ رجاءً أجيبي على جميع الحقول."
+        return "⚠️ رجاءً أجيبي على جميع الأسئلة أولاً."
+
     prompt = (
-        f"اكتب جملة عربية قصيرة وملهمة تصف مستقبل شخص يريد {future}، "
+        f"اكتبي بيت شعر أو عبارة عربية قصيرة وملهمة تصف مستقبل شخص يريد {future}، "
         f"ويؤمن بـ {value}، ويمتلك مهارة {skill}. "
-        "لتكن الجملة إنسانية وشاعرية ومليئة بالأمل."
+        "لتكن شاعرية ومفعمة بالأمل والجمال."
     )
+
     try:
         result = generator(
             prompt,
             max_new_tokens=60,
             temperature=0.8,
-            top_p=0.9
+            top_p=0.9,
+            repetition_penalty=1.1
         )
         text = result[0]["generated_text"].strip()
         if text.startswith(prompt):
             text = text[len(prompt):].strip()
-        return text or "✨ لم يولّد النموذج نصًا، جرّبي إدخالًا أبسط."
+        return text or "✨ النموذج لم يولّد نتيجة، جرّبي كلمات أبسط."
     except Exception as e:
         return f"⚠️ حدث خطأ أثناء التوليد: {e}"
 
@@ -45,11 +37,11 @@ demo = gr.Interface(
     inputs=[
         gr.Textbox(label="🎯 حلمك المستقبلي"),
         gr.Textbox(label="💡 الأثر الذي تريدين تركه"),
-        gr.Textbox(label="⚙️ مهارتك المميزة"),
+        gr.Textbox(label="⚙️ مهارتك المميزة")
     ],
-    outputs=gr.Textbox(label="🌟 رؤيتك المستقبلية", lines=3),
-    title="ذكاء يصمم مستقبلك بالعربية",
-    description="نموذج عربي متقدّم يكتب جملة ملهمة بناءً على قيمك وأحلامك.",
+    outputs=gr.Textbox(label="✨ رؤيتك المستقبلية", lines=4),
+    title="ذكاء يصمم مستقبلك - بإحساس شعري 🎨",
+    description="نموذج عربي شعري يكتب رؤيتك المستقبلية بأسلوب فني مفعم بالأمل.",
     theme="soft"
 )