Update app.py

app.py

@@ -8,33 +8,32 @@ import uuid
 from typing import Tuple
 import numpy as np
 
+# Function to save an image with a unique name
 def save_image(img):
     unique_name = str(uuid.uuid4()) + ".png"
     img.save(unique_name)
     return unique_name
 
+# Function to handle seed randomization
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     return seed
 
+# Maximum seed value for 32-bit integer
 MAX_SEED = np.iinfo(np.int32).max
 
-if not torch.cuda.is_available():
-    DESCRIPTIONz += "\n<p>⚠️Running on CPU, This may not work on CPU.</p>"
-
+# Load the diffusion model
 base_model = "black-forest-labs/FLUX.1-dev"
 pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
 
-#lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
-#trigger_word = "Super Realism"  # Leave trigger_word blank if not used.
-
-lora_repo = "strangerzonehf/FallenArt-Flux"
-trigger_word = "Fallen Art"  # Leave trigger_word blank if not used.
+lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
+trigger_word = "Super Realism"  # Leave blank if not used
 
 pipe.load_lora_weights(lora_repo)
 pipe.to("cuda")
 
+# Define style options
 style_list = [
     {
         "name": "3840 x 2160",
@@ -55,13 +54,14 @@ style_list = [
 ]
 
 styles = {k["name"]: k["prompt"] for k in style_list}
-
 DEFAULT_STYLE_NAME = "3840 x 2160"
 STYLE_NAMES = list(styles.keys())
 
+# Apply selected style to the prompt
 def apply_style(style_name: str, positive: str) -> str:
     return styles.get(style_name, styles[DEFAULT_STYLE_NAME]).replace("{prompt}", positive)
 
+# Image generation function with Spaces GPU support
 @spaces.GPU(duration=60, enable_queue=True)
 def generate(
     prompt: str,
@@ -74,7 +74,6 @@ def generate(
     progress=gr.Progress(track_tqdm=True),
 ):
     seed = int(randomize_seed_fn(seed, randomize_seed))
-
     positive_prompt = apply_style(style_name, prompt)
     
     if trigger_word:
@@ -93,6 +92,7 @@ def generate(
     print(image_paths)
     return image_paths, seed
 
+# Example prompts
 examples = [
     "Super Realism, High-resolution photograph, woman, UHD, photorealistic, shot on a Sony A7III --chaos 20 --ar 1:2 --style raw --stylize 250",
     "Woman in a red jacket, snowy, in the style of hyper-realistic portraiture, caninecore, mountainous vistas, timeless beauty, palewave, iconic, distinctive noses --ar 72:101 --stylize 750 --v 6",
@@ -100,11 +100,17 @@ examples = [
     "Super-realism, Purple Dreamy, a medium-angle shot of a young woman with long brown hair, wearing a pair of eye-level glasses, stands in front of a backdrop of purple and white lights. The womans eyes are closed, her lips are slightly parted, as if she is looking up at the sky. Her hair is cascading over her shoulders, framing her face. She is wearing a sleeveless top, adorned with tiny white dots, and a gold chain necklace around her neck. Her left earrings are dangling from her ears, adding a pop of color to the scene."
 ]
 
+# CSS to center the UI and style components
 css = '''
-.gradio-container{max-width: 888px !important}
-h1{text-align:center}
+.gradio-container {
+    max-width: 888px !important;
+    margin: 0 auto !important;
+}
+h1 {
+    text-align: center;
+}
 footer {
-    visibility: hidden
+    visibility: hidden;
 }
 .submit-btn {
     background-color: #e34949 !important;
@@ -115,6 +121,7 @@ footer {
 }
 '''
 
+# Gradio interface
 with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
     with gr.Row():
         with gr.Column(scale=1):
@@ -125,7 +132,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                 placeholder="Enter your prompt",
                 container=False,
             )
-            run_button = gr.Button("Generate as ( 768 x 1024 )🤗", scale=0, elem_classes="submit-btn")
+            run_button = gr.Button("Generate Image", scale=0, elem_classes="submit-btn")
             
             with gr.Accordion("Advanced options", open=True, visible=True):
                 seed = gr.Slider(