Update app.py

app.py

@@ -5,6 +5,8 @@ import os
 from datetime import datetime
 import random
 from transformers import pipeline
+import torch
+from diffusers import FluxPipeline
 
 #----------Start of theme----------
 theme = gr.themes.Soft(
@@ -291,12 +293,39 @@ css = """
 with gr.Blocks(theme=theme, css=css) as app:
     gr.HTML("<center><h6>🎨 Image Studio</h6></center>")
     with gr.Tab("Text to Image"):
-        
         #gr.HTML("<center><b>Flux</b></center>")
         #gr.load("models/XLabs-AI/flux-RealismLora")
         #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")
+        #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
+    
+    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")
+    
 
     
     with gr.Tab("Flip Image"):