Update app.py

app.py

@@ -298,33 +298,33 @@ with gr.Blocks(theme=theme, css=css) as app:
         #gr.load("models/digiplay/AnalogMadness-realistic-model-v7")
         #gr.load("stabilityai/stable-diffusion-3-medium-diffusers")
         #model=gr.load("models/dvyio/flux-lora-film-noir")
-    base_model_id = "Freepik/flux.1-lite-8B-alpha"
-    torch_dtype = torch.bfloat16
-    device = "cuda"
-
-    # Load the pipe
-    model_id = "Freepik/flux.1-lite-8B-alpha"
-    pipe = FluxPipeline.from_pretrained(
-        model_id, torch_dtype=torch_dtype
-    ).to(device)
-    
-    # Inference
-    prompt = "A close-up image of a green alien with fluorescent skin in the middle of a dark purple forest"
-    
-    guidance_scale = 3.5  # Keep guidance_scale at 3.5
-    n_steps = 28
-    seed = 11
+        base_model_id = "Freepik/flux.1-lite-8B-alpha"
+        torch_dtype = torch.bfloat16
+        device = "cuda"
     
-    with torch.inference_mode():
-        image = pipe(
-            prompt=prompt,
-            generator=torch.Generator(device="cpu").manual_seed(seed),
-            num_inference_steps=n_steps,
-            guidance_scale=guidance_scale,
-            height=1024,
-            width=1024,
-        ).images[0]
-    image.save("output.png")
+        # Load the pipe
+        model_id = "Freepik/flux.1-lite-8B-alpha"
+        pipe = FluxPipeline.from_pretrained(
+            model_id, torch_dtype=torch_dtype
+        ).to(device)
+        
+        # Inference
+        prompt = "A close-up image of a green alien with fluorescent skin in the middle of a dark purple forest"
+        
+        guidance_scale = 3.5  # Keep guidance_scale at 3.5
+        n_steps = 28
+        seed = 11
+        
+        with torch.inference_mode():
+            image = pipe(
+                prompt=prompt,
+                generator=torch.Generator(device="cpu").manual_seed(seed),
+                num_inference_steps=n_steps,
+                guidance_scale=guidance_scale,
+                height=1024,
+                width=1024,
+            ).images[0]
+        image.save("output.png")