Update app.py

app.py

@@ -297,34 +297,10 @@ with gr.Blocks(theme=theme, css=css) as app:
         #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")
-        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
+        model=gr.load("models/dvyio/flux-lora-film-noir")
+        base_model_id = "dvyio/flux-lora-film-noir"
         
-        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")
+