Spaces:
Sleeping
Sleeping
File size: 3,083 Bytes
c62628b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 |
# @title chạy với giao diện Gradio (load mode từ local)
import gradio as gr
from diffusers import StableDiffusionPipeline
from googletrans import Translator
import torch
# Model initialization
def load_model(model_path):
pipe = StableDiffusionPipeline.from_single_file(model_path, safety_checker=None, requires_safety_checker=False, torch_dtype=torch.float16)
return pipe.to("cuda")
model_path = r'D:\Truong_apacons_2023\EasySD\models\stable-diffusion\realisticVisionV51_v51VAE.safetensors'
pipe = load_model(model_path)
translator = Translator()
#mô tả tiêu cực
negative_prompt = "disfigured, ugly, verybadimagenegative_v1.3, ng_deepnegative_v1_75t, (ugly face:0.8),cross-eyed,sketches, (worst quality:2), (low quality:2), (normal quality:2), lowres, normal quality, ((monochrome)), ((grayscale)), bad anatomy, DeepNegative, facing away, tilted head, {Multiple people}, lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worstquality, low quality, normal quality, jpegartifacts, signature, watermark, username, blurry..."
def generate(prompt, mau_nuoc, render, height=512, width=768):
height = int(height)
width = int(width)
if mau_nuoc:
prompt = "Tranh Màu nước, " + prompt
if render:
prompt = "Octain render and Vray render and Corona render 3d model, fog effect, perfect perspective, perfect edges, " + prompt
# Xoá từ "realistic", "photographic" và "photorealistic"
prompt = prompt.replace("realistic", "").replace("photographic", "").replace("photorealistic", "")
# Translate
translated_prompt = translator.translate(prompt).text
# Hàm sinh ảnh
images = pipe(
prompt=translated_prompt,
height=height,
width=width,
num_inference_steps=50,
guidance_scale=7,
negative_prompt=negative_prompt,
).images
image = images[0]
return image
Ví_dụ = [
["Chân dung chi tiết cao của một cô gái trẻ xinh đẹp với mái tóc trắng và đôi mắt sắc nét, 35 mm, 8k, sắc nét"],
["Cô gái xinh đẹp mặc váy màu vàng chơi guitar, đính kim tuyến lộng lẫy, hình ảnh sắc nét, không bị mờ"],
["Ngôi nhà màu vàng trong rừng, ánh sáng mặt trời tuyệt đẹp"],
["A cat playing with a ball of yarn"],
["Chú chó chạy xe đạp màu hồng"],
["Bình minh trên bờ biển"]
]
iface = gr.Interface(
fn=generate,
inputs=[
gr.inputs.Textbox(),
gr.inputs.Checkbox(label="Màu nước"),
gr.inputs.Checkbox(label="Render"),
gr.inputs.Number(label="Kích thước ảnh - Chiều cao", default=512),
gr.inputs.Number(label="Kích thước ảnh - Chiều rộng", default=768)
],
outputs="image",
title="Sáng tạo hình ảnh",
description="Mô tả về chương trình<br>Hãy viết mô tả mà bạn mong muốn vào ô nhập prompt và bấm vào nút Submit",
examples=Ví_dụ
)
iface.launch(share=True, debug=True)
|