IdeaSpark / app.py
Abdallah-Ai's picture
Create app.py
afc9d60 verified
raw
history blame
1.08 kB
import streamlit as st
from transformers import pipeline
st.title("مولد أفكار المحتوى الإبداعي")
st.header("أدخل فكرتك وسأقوم بتوليد أفكار جذابة للمحتوى الخاص بك")
# هذا نموذج لغوي عربي قوي من Hugging Face
generator = pipeline('text-generation', model='aubmindlab/aragpt2-large')
user_input = st.text_input("اكتب فكرتك هنا:")
if user_input:
if st.button("توليد الأفكار"):
with st.spinner('جاري توليد الأفكار...'):
generated_text = generator(
f"أفكار لمحتوى عن {user_input}:",
max_length=150,
num_return_sequences=3,
pad_token_id=generator.tokenizer.eos_token_id
)
st.success("تم توليد الأفكار بنجاح!")
for i, idea in enumerate(generated_text):
st.write(f"**فكرة {i+1}:** {idea['generated_text'].replace(f'أفكار لمحتوى عن {user_input}:', '', 1).strip()}")