Update app.py

app.py

@@ -11,7 +11,7 @@ pipe = pipe.to(device)
 
 def genie (prompt, scale, steps, seed):
      generator = torch.Generator(device=device).manual_seed(seed)
-     images = pipe(prompt, width=768, height=768, num_inference_steps=steps, guidance_scale=scale, num_images_per_prompt=1, generator=generator).images[0]
+     images = pipe(prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
      return images
     
 gr.Interface(fn=genie, inputs=[gr.Textbox(label='What you want the AI to generate. 77 Token Limit.'),