Update app.py

app.py

@@ -35,8 +35,8 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
         seed = random.randint(0, MAX_SEED)
     return seed
 
-@spaces.GPU(queue=False,duration=30)
-def generate(
+@spaces.GPU
+def infer(
     prompt: str,
     negative_prompt: str = "",
     use_negative_prompt: bool = False,
@@ -49,7 +49,6 @@ def generate(
     use_resolution_binning: bool = True,
     progress=gr.Progress(track_tqdm=True),
 ):
-    pipe.to(device)
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)
     image = pipe(
@@ -148,7 +147,7 @@ with gr.Blocks(css=css) as demo:
         examples=examples,
         inputs=prompt,
         outputs=[result, seed],
-        fn=generate,
+        fn=infer,
         cache_examples=CACHE_EXAMPLES,
     )
 
@@ -164,7 +163,7 @@ with gr.Blocks(css=css) as demo:
             negative_prompt.submit,
             run_button.click,
         ],
-        fn=generate,
+        fn=infer,
         inputs=[
             prompt,
             negative_prompt,