StableDiffusion / app.py
malekradwan130's picture
Update app.py
4defb62 verified
raw
history blame contribute delete
741 Bytes
from huggingface_hub import login
from diffusers import DiffusionPipeline
import gradio as gr
import os
# تسجيل الدخول باستخدام الرمز
login(token=os.environ.get("HF_TOKEN"))
# تحميل النموذج
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large-turbo")
# دالة لإنشاء الصور
def generate_image(prompt):
image = pipe(prompt).images[0]
return image
# واجهة Gradio
iface = gr.Interface(
fn=generate_image,
inputs="text",
outputs="image",
title="Stable Diffusion 3.5 Large Turbo",
description="أدخل وصفًا لإنشاء صورة باستخدام Stable Diffusion 3.5 Large Turbo."
)
# تشغيل الواجهة
iface.launch()