from diffusers import StableDiffusionPipeline
import torch

def generate_realistic(prompt: str):
    model_id = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
    pipe = StableDiffusionPipeline.from_pretrained(
        model_id,
        torch_dtype=torch.float16,
        use_safetensors=False,
        # variant="fp16"
    ).to("cuda")
    
    image = pipe(prompt=prompt).images[0]
    return image

if __name__ == "__main__":
    prompt = '그녀를 바라보는 한 남자의 야망'
    image = generate_realistic(prompt)
    image.save("output_realistic.png")
    print("✅ 저장 완료: output_realistic.png")
    image.show()