Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import pipeline | |
| # تحميل نموذج GPT-Neo 125M | |
| generator = pipeline("text-generation", model="EleutherAI/gpt-neo-125M", device=-1) | |
| # ذاكرة افتراضية للبوت | |
| memory = [] | |
| def generate_post(): | |
| # توليد نص قصير | |
| prompt = "Write a short English text (<=100 chars):" | |
| output = generator(prompt, max_new_tokens=30, do_sample=True, temperature=1.0) | |
| text = output[0]['generated_text'].strip() | |
| # قص النص إذا كان طويل | |
| if len(text) > 100: | |
| text = text[:100] | |
| # إضافة للذاكرة | |
| memory.append(text) | |
| return text | |
| def stop_and_clear(): | |
| memory.clear() | |
| return "🛑 Bot stopped and memory cleared!" | |
| with gr.Blocks() as demo: | |
| gr.Markdown("## 🤖 Auto-writing Bot") | |
| post_box = gr.Textbox(label="Generated Post", lines=2) | |
| generate_btn = gr.Button("Generate Post") | |
| stop_btn = gr.Button("Stop & Clear Memory") | |
| generate_btn.click(fn=generate_post, outputs=post_box) | |
| stop_btn.click(fn=stop_and_clear, outputs=post_box) | |
| demo.launch() |