|
|
import os |
|
|
import random |
|
|
import gradio as gr |
|
|
import torch |
|
|
|
|
|
|
|
|
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" |
|
|
|
|
|
|
|
|
model_id = "stabilityai/sd-turbo" |
|
|
device = "cuda" if torch.cuda.is_available() else "cpu" |
|
|
pipe = None |
|
|
|
|
|
styles = [ |
|
|
"punk fashion portrait, neon jacket, futuristic hairstyle, urban background", |
|
|
"futuristic metallic outfit, cyberpunk city, glowing textures, sci-fi theme", |
|
|
"vintage 80s fashion, colorful retro lighting, analog grain effect", |
|
|
"haute couture studio photoshoot, elegant lighting, fashion magazine", |
|
|
"streetwear outfit, graffiti wall, baggy hoodie, bold pose", |
|
|
"anime inspired outfit, colorful clothing, expressive character art", |
|
|
"fantasy armor, cinematic lighting, ethereal atmosphere", |
|
|
"minimalist pastel outfit, soft background, calm mood" |
|
|
] |
|
|
|
|
|
|
|
|
def load_model(): |
|
|
global pipe |
|
|
if pipe is None: |
|
|
from diffusers import StableDiffusionPipeline |
|
|
pipe = StableDiffusionPipeline.from_pretrained( |
|
|
model_id, |
|
|
torch_dtype=torch.float16 if device == "cuda" else torch.float32 |
|
|
).to(device) |
|
|
return pipe |
|
|
|
|
|
def restyle(image, custom_prompt=None): |
|
|
pipe = load_model() |
|
|
chosen_style = custom_prompt.strip() if custom_prompt else random.choice(styles) |
|
|
prompt = f"A portrait of the same person wearing {chosen_style}, ultra realistic, detailed face, professional studio lighting" |
|
|
result = pipe(prompt=prompt, image=image, guidance_scale=1.5).images[0] |
|
|
return result, f"Kullanılan Stil: {chosen_style}" |
|
|
|
|
|
|
|
|
with gr.Blocks(theme=gr.themes.Monochrome(), css=""" |
|
|
#title {text-align:center; font-size:2em; font-weight:bold; margin-bottom:10px;} |
|
|
#desc {text-align:center; font-size:1.1em; color:#cccccc; margin-bottom:25px;} |
|
|
footer {visibility:hidden} |
|
|
""") as demo: |
|
|
|
|
|
gr.HTML("<div id='title'>🧥 AI Stylist - Turbo Sürüm ⚡</div>") |
|
|
gr.HTML("<div id='desc'>Yapay zeka senin fotoğrafına rastgele bir moda stili uygulasın.<br>GPU varsa 5 saniyede, yoksa biraz sabır 🙂</div>") |
|
|
|
|
|
with gr.Row(): |
|
|
with gr.Column(scale=1): |
|
|
img_input = gr.Image(type="pil", label="📸 Fotoğrafını Yükle") |
|
|
txt_style = gr.Textbox(label="🪄 Özel Stil (boş bırak = rastgele)") |
|
|
btn_generate = gr.Button("🎲 Stil Üret / Yeniden Oluştur") |
|
|
with gr.Column(scale=1): |
|
|
img_output = gr.Image(label="🧢 Yeni Stil") |
|
|
txt_output = gr.Textbox(label="Stil Bilgisi", interactive=False) |
|
|
|
|
|
btn_generate.click(fn=restyle, inputs=[img_input, txt_style], outputs=[img_output, txt_output]) |
|
|
|
|
|
gr.HTML("<div style='text-align:center; color:#888; margin-top:20px;'>✨ Katake tarafından geliştirildi · AI Stylist Turbo Sürüm</div>") |
|
|
|
|
|
demo.launch() |
|
|
|
