Upload app.py

app.py

@@ -33,7 +33,7 @@ pipe.unet.to(torch.float16)
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1216
     
-@spaces.GPU(duration=10)
+@spaces.GPU()
 def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
     # Check and truncate prompt if too long (CLIP can only handle 77 tokens)
     if len(prompt.split()) > 60:  # Rough estimate to avoid exceeding token limit
@@ -66,7 +66,7 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
 css = """
 #col-container {
     margin: 0 auto;
-    max-width: 520px;
+    max-width: 1024px;
 }
 """