Spaces:
Sleeping
Sleeping
File size: 1,095 Bytes
985bbd5 51b7946 38dc6ab 86daa57 38dc6ab 86daa57 38dc6ab 86daa57 b9a93f6 86daa57 b9a93f6 86daa57 759c944 86daa57 985bbd5 86daa57 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | 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() |