Update app.py

app.py

@@ -19,19 +19,30 @@ import gradio as gr
 import diffusers
 import torch
 
-# Define the function to generate an image from text using the Liberata/illustrious-xl-v1.0 model
+# Automatically detect if CUDA is available and set the device accordingly
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# Load the model from the specified file outside the function to ensure better performance
+model = diffusers.StableDiffusionXLPipeline.from_single_file(
+    "https://huggingface.co/Liberata/illustrious-xl-v1.0/blob/main/Illustrious-XL-v1.0.safetensors",
+    torch_dtype=torch.float16
+).to(device)  # Move the model to the detected device
+
+# Define the function to generate an image from text using the pre-loaded model
 def generate_image(prompt, guidance_scale, num_inference_steps):
-    # Load the model from the specified file
-    model = diffusers.StableDiffusionXLPipeline.from_pretrained(
-        "Liberata/illustrious-xl-v1.0",
-        torch_dtype=torch.float16
-    )
     # Generate the image from the text prompt with the specified parameters
     image = model(prompt, guidance_scale=guidance_scale, num_inference_steps=num_inference_steps).images[0]
     return image
 
 # Create a Gradio interface for the text-to-image generation
 with gr.Blocks() as demo:
+        # Add some documentation using HTML
+    gr.HTML("""
+    <h2> 😊 Text-to-Image Generation with Liberata/illustrious-xl-v1.0</h2>
+    <p>This app allows you to generate images from text prompts using the Liberata/illustrious-xl-v1.0 model.</p>
+    <p>Simply enter a text prompt, adjust the parameters, and click the 'Generate Image' button to see the generated image.</p>
+    """)
+    
     # Create a two-column layout
     with gr.Row():
         with gr.Column():
@@ -46,13 +57,6 @@ with gr.Blocks() as demo:
             # Create an image output to display the generated image
             image_output = gr.Image(label="Generated Image")
     
-    # Add some documentation using HTML
-    gr.HTML("""
-    <h2>Text-to-Image Generation with Liberata/illustrious-xl-v1.0</h2>
-    <p>This app allows you to generate images from text prompts using the Liberata/illustrious-xl-v1.0 model.</p>
-    <p>Simply enter a text prompt, adjust the parameters, and click the 'Generate Image' button to see the generated image.</p>
-    """)
-    
     # Create a button to trigger the image generation
     generate_button = gr.Button("Generate Image")
     
@@ -61,6 +65,8 @@ with gr.Blocks() as demo:
     
     # Add examples
     examples = [
+        ["1boy, Digital anime-style ALBEDO has light gray, messy hair, blue eyes. Shy look, in white & blue hoodie, at cozy café table with food. Shelf background"],
+        ["1boy ,Digital anime-style CHONGYUN has shiny silver hair, green eyes. Cheerful expression, in bright blue T-shirt, at luxurious restaurant with Belgian chocolates. Elegant crystal chandelier background."],
         ["A fox drinking tea under a cherry blossom tree, anime style, 4K resolution"],
         ["Night view of a futuristic city, cyberpunk style, neon lights"],
         ["Medieval knight battling a dragon, epic scene, oil painting texture"]