Update app.py

app.py

@@ -3,8 +3,7 @@ import numpy as np
 import random
 import spaces
 import torch
-from diffusers import  DiffusionPipeline, FlowMatchEulerDiscreteScheduler
-from transformers import CLIPTextModel, CLIPTokenizer,T5EncoderModel, T5TokenizerFast
+from diffusers import DiffusionPipeline
 from PIL import Image
 import uuid
 from typing import Tuple
@@ -12,7 +11,7 @@ from typing import Tuple
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype).to(device)
+pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
@@ -192,7 +191,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                     minimum=1,
                     maximum=50,
                     step=1,
-                    value=28,
+                    value=4,
                 )
                 
         gr.Examples(