Update app.py

app.py

@@ -297,11 +297,45 @@ 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")
-        gr.Interface.load("models/dvyio/flux-lora-film-noir")
-        base_model_id = "dvyio/flux-lora-film-noir"
+        def query(prompt, is_negative=False, steps=28, cfg_scale=3.5, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
+            if prompt == "" or prompt == None:
+                return None
+    # Prepare the payload for the API call, including width and height
+    payload = {
+        "inputs": prompt,
+        "is_negative": is_negative,
+        "steps": steps,
+        "cfg_scale": cfg_scale,
+        "seed": seed if seed != -1 else random.randint(1, 1000000000),
+        "strength": strength,
+        "parameters": {
+            "width": width,  # Pass the width to the API
+            "height": height  # Pass the height to the API
+        }
+    }
+        model=("models/dvyio/flux-lora-film-noir")#base_model_id = "dvyio/flux-lora-film-noir"
+        with gr.Row(model):
+            text_prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt here", lines=2, elem_id="prompt-text-input")
         
- 
-    
+        with gr.Row(model):
+            with gr.Accordion("Advanced Settings", open=False):
+                negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image", value="((((out of frame))), deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, misspellings, typos", lines=3, elem_id="negative-prompt-text-input")
+            with gr.Row(model):
+                width = gr.Slider(label="Width", value=1024, minimum=64, maximum=1216, step=32)
+                height = gr.Slider(label="Height", value=1024, minimum=64, maximum=1216, step=32)
+                steps = gr.Slider(label="Sampling steps", value=28, minimum=1, maximum=100, step=1)
+                cfg = gr.Slider(label="CFG Scale", value=3.5, minimum=1, maximum=20, step=0.5)
+                method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "DPM Fast" "Euler", "Euler a", "Euler+beta", "Heun", "DDIM", "PLMS", "UniPC"])
+                strength = gr.Slider(label="Strength", value=0.7, minimum=0, maximum=1, step=0.001)
+                seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1)
+            with gr.Row(model):
+                with gr.Accordion("Seed", open=False):
+                    seed_output = gr.Textbox(label="Seed Used", show_copy_button = True, elem_id="seed-output")
+            with gr.Row(model):
+                text_button = gr.Button("Run", variant='primary', elem_id="gen-button")            
+            with gr.Row(model):
+                image_output = gr.Image(type="pil", label="Image Output", format="png", elem_id="gallery")
+text_button.click(query, inputs=[custom_lora, text_prompt, negative_prompt, steps, cfg, method, seed, strength, width, height], outputs=[image_output, seed_output])
 
     
     with gr.Tab("Flip Image"):