Update app.py

app.py

@@ -6,10 +6,10 @@ from diffusers import DiffusionPipeline
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "stabilityai/sdxl-turbo" #Replace to the model you would like to use
+model_repo_id = "black-forest-labs/FLUX.1-dev" #Replace to the model you would like to use
 
 if torch.cuda.is_available():
-    torch_dtype = torch.float16
+    torch_dtype = torch.bfloat16
 else:
     torch_dtype = torch.float32
 
@@ -60,36 +60,35 @@ with gr.Blocks(css=css) as demo:
         # FLUX.1 [dev] with CFG (and negative prompts)
         """)
         
+        #with gr.Row():
         with gr.Row():
-            with gr.Row():
-                prompt = gr.Text(
-                    label="Prompt",
-                    show_label=False,
-                    max_lines=1,
-                    placeholder="Enter your prompt",
-                    container=False,
-                )
-                negative_prompt = gr.Text(
-                    label="Negative prompt",
-                    max_lines=1,
-                    placeholder="Enter a negative prompt",
-                    visible=False,
-                )
-            with gr.Row():
-                guidance_scale = gr.Slider(
-                    label="Guidance scale",
-                    minimum=0.0,
-                    maximum=10.0,
-                    step=0.1,
-                    value=1.0, #Replace with defaults that work for your model
-                )
-                true_guidance = gr.Slider(
-                    label="True CFG scale",
-                    minimum=0.0,
-                    maximum=10.0,
-                    step=0.1,
-                    value=5.0, #Replace with defaults that work for your model
-                )
+            prompt = gr.Text(
+                label="Prompt",
+                show_label=False,
+                max_lines=1,
+                placeholder="Enter your prompt",
+                container=False,
+            )
+            negative_prompt = gr.Text(
+                label="Negative prompt",
+                max_lines=1,
+                placeholder="Enter a negative prompt",
+            )
+        with gr.Row():
+            guidance_scale = gr.Slider(
+                label="Guidance scale",
+                minimum=0.0,
+                maximum=10.0,
+                step=0.1,
+                value=1.0, #Replace with defaults that work for your model
+            )
+            true_guidance = gr.Slider(
+                label="True CFG scale",
+                minimum=0.0,
+                maximum=10.0,
+                step=0.1,
+                value=5.0, #Replace with defaults that work for your model
+            )
             run_button = gr.Button("Run", scale=0)
         
         result = gr.Image(label="Result", show_label=False)