import streamlit as st
from transformers import pipeline

# تحميل النموذج باستخدام Hugging Face
generator = pipeline("text-generation", model="inceptionai/jais-13b")

# عنوان التطبيق
st.title("تطبيق توليد النصوص باستخدام Jais-13B")

# إدخال من المستخدم
input_text = st.text_input("اكتب نصًا لتوليد نص مشابه له:")

# عندما يضغط المستخدم على زر التوليد
if input_text:
    # استخدام النموذج لتوليد النصوص
    output_text = generator(input_text, max_length=100, num_return_sequences=1)
    
    # عرض النص المُولد
    st.write("النص المُولد:")
    st.write(output_text[0]['generated_text'])