Update app.py

app.py

@@ -32,10 +32,10 @@ JS = """function () {
 if torch.cuda.is_available():
     pipe = FluxPipeline.from_pretrained(
         model, 
-        torch_dtype=torch.bfloat16)
+        torch_dtype=torch.bfloat16).to("cuda")
 
 # Function 
-@spaces.GPU()
+@spaces.GPU(duration=120)
 def generate_image(
     prompt,
     width=1024,
@@ -125,10 +125,10 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
                     minimum=1,
                     maximum=100,
                     step=1,
-                    value=50,
+                    value=30,
                 )
                 seed = gr.Slider(
-                    label="Seed(-1 Random)",
+                    label="Seeds",
                     minimum=-1,
                     maximum=MAX_SEED,
                     step=1,