Update app.py

app.py

@@ -11,11 +11,11 @@ upscaler = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-
 upscaler = upscaler.to(device)
 pipe = pipe.to(device)
 
-def genie (prompt, scale, steps, seed):
+def genie (Prompt, scale, steps, seed):
      generator = torch.Generator(device=device).manual_seed(seed)
      #images = pipe(prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
-     low_res_latents = pipe(prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator, output_type="latent").images
-     upscaled_image = upscaler(prompt=prompt, image=low_res_latents, num_inference_steps=20, guidance_scale=0, generator=generator).images[0]
+     low_res_latents = pipe(Prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator, output_type="latent").images
+     upscaled_image = upscaler(prompt='', image=low_res_latents, num_inference_steps=5, guidance_scale=0, generator=generator).images[0]
      return upscaled_image
     
 gr.Interface(fn=genie, inputs=[gr.Textbox(label='What you want the AI to generate. 77 Token Limit.'),