Death-1 / app.py
Skydata001's picture
Update app.py
86daa57 verified
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()