import spaces import gradio as gr import numpy as np import random import os from dotenv import load_dotenv from diffusers import AutoPipelineForText2Image import torch from huggingface_hub import snapshot_download load_dotenv() snapshot_download(repo_id="Roomie/xavyy", cache_dir='./') pipeline = AutoPipelineForText2Image.from_pretrained( 'black-forest-labs/FLUX.1-schnell', torch_dtype=torch.bfloat16).to('cuda') pipeline.load_lora_weights('Roomie/xavyy', weight_name='xavyy.safetensors') MAX_SEED = np.iinfo(np.int32).max MAX_IMAGE_SIZE = 1024 @spaces.GPU def infer(prompt, height, width): image = pipeline( prompt=prompt, # guidance_scale=guidance_scale, # num_inference_steps=num_inference_steps, width=width, height=height, # generator=generator ).images[0] return image examples = [ "Xavy, a virtual content creator, is in a high-tech futuristic studio filled with holographic screens and cutting-edge gadgets. He’s presenting the latest smartphone technology, wearing a sleek tech-inspired outfit with neon accents. The background features floating data, robots assisting him, and advanced digital tools. His facial expression is enthusiastic as he explains the potential of artificial intelligence in smartphones. The atmosphere is dynamic and full of futuristic energy. Negative prompt: avoid multiple versions of Xavy, avoid distorted facial features, malformed hands, excessive or broken gadgets, unrealistic proportions in the body or technology, extra limbs.", "Xavy stands on a stage at a technology innovation conference, speaking passionately about the future of AI in smartphones. Behind him, a massive screen displays 3D holographic models of a cutting-edge phone design. The audience is captivated as he gestures towards the hologram, explaining how AI will revolutionize user interaction. He’s wearing a sleek black outfit with a futuristic smartwatch, and the lighting is focused on him while the background is filled with technological elements like drones and digital billboards. Negative prompt: avoid duplicated Xavy figures, warped or incomplete body parts, malformed facial expressions, extra gadgets or overlapping elements, unnatural lighting, broken equipment, unrealistic audience features." ] css = """ #col-container { margin: 0 auto; max-width: 640px; } """ with gr.Blocks(css=css) as demo: with gr.Column(elem_id="col-container"): gr.Markdown(f""" # Text-to-Image Gradio Template """) with gr.Row(): prompt = gr.TextArea( label="Prompt", show_label=False, max_lines=1, placeholder="Enter your prompt", container=False, ) run_button = gr.Button("Run", scale=0) result = gr.Image(label="Result", show_label=False) with gr.Accordion("Advanced Settings", open=False): negative_prompt = gr.TextArea( label="Negative prompt", max_lines=1, placeholder="Default Negative prompt: avoid multiple versions of Xavy, avoid distorted facial features, malformed hands, excessive or broken gadgets, unrealistic proportions in the body or technology, extra limbs", visible=False, ) seed = gr.Slider( label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, visible=False ) randomize_seed = gr.Checkbox(label="Randomize seed", value=True) with gr.Row(): width = gr.Slider( label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=400, # Replace with defaults that work for your model ) height = gr.Slider( label="Height", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=600, # Replace with defaults that work for your model ) with gr.Row(): guidance_scale = gr.Slider( label="Guidance scale", minimum=0.0, maximum=10.0, step=0.1, value=0.0, # Replace with defaults that work for your model visible=False ) num_inference_steps = gr.Slider( label="Number of inference steps", minimum=1, maximum=50, step=1, value=2, # Replace with defaults that work for your model visible=False ) gr.Examples( examples=examples, inputs=[prompt] ) gr.on( triggers=[run_button.click, prompt.submit], fn=infer, inputs=[prompt, height, width], outputs=[result] ) demo.queue().launch(debug=True, auth=(os.environ.get( 'SPACE_USERNAME'), os.environ.get('SPACE_PASSWORD')))