Update app.py

app.py

@@ -10,12 +10,13 @@ dtype = torch.float16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load the diffusion pipeline
-pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
+pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype)
+pipe = pipe.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
-def infer(prompt, negative_prompt="", seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, guidance_scale=7.5, progress=gr.Progress(track_tqdm=True)):
+def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, guidance_scale=7.5, progress=gr.Progress(track_tqdm=True)):
     start_time = time.time()
 
     if width > MAX_IMAGE_SIZE or height > MAX_IMAGE_SIZE:
@@ -28,7 +29,6 @@ def infer(prompt, negative_prompt="", seed=42, randomize_seed=False, width=1024,
     try:
         image = pipe(
             prompt=prompt,
-            negative_prompt=negative_prompt,
             width=width,
             height=height,
             num_inference_steps=num_inference_steps,
@@ -45,9 +45,9 @@ def infer(prompt, negative_prompt="", seed=42, randomize_seed=False, width=1024,
     return image, seed, None
 
 examples = [
-    ["a tiny astronaut hatching from an egg on the moon", "blurry, low quality"],
-    ["a cat holding a sign that says hello world", "dog, text, writing"],
-    ["an anime illustration of a wiener schnitzel", "realistic, photograph"],
+    ["a tiny astronaut hatching from an egg on the moon"],
+    ["a cat holding a sign that says hello world"],
+    ["an anime illustration of a wiener schnitzel"],
 ]
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -64,11 +64,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 placeholder="Enter your prompt",
                 lines=3
             )
-            negative_prompt = gr.Textbox(
-                label="Negative Prompt",
-                placeholder="Enter things to avoid in the image",
-                lines=2
-            )
             run_button = gr.Button("Generate Image", variant="primary")
 
         with gr.Column(scale=2):
@@ -89,7 +84,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     gr.Examples(
         examples=examples,
-        inputs=[prompt, negative_prompt],
+        inputs=[prompt],
         outputs=[result, seed_output],
         fn=infer,
         cache_examples=True
@@ -97,7 +92,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     run_button.click(
         fn=infer,
-        inputs=[prompt, negative_prompt, seed, randomize_seed, width, height, num_inference_steps, guidance_scale],
+        inputs=[prompt, seed, randomize_seed, width, height, num_inference_steps, guidance_scale],
         outputs=[result, seed_output]
     )