Update app.py

Removed unneeded and experimental settings. Now passing Prompt and Negative Prompt directly to Upscaler

app.py

@@ -11,11 +11,11 @@ upscaler = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-
 upscaler = upscaler.to(device)
 pipe = pipe.to(device)
 
-def genie (Prompt, negative_prompt, height, width, scale, steps, seed, upscale, upscale_prompt, upscale_neg, upscale_scale, upscale_steps):
+def genie (Prompt, negative_prompt, height, width, scale, steps, seed, upscale):
     generator = torch.Generator(device=device).manual_seed(seed)
     if upscale == "Yes":
         low_res_latents = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, generator=generator, output_type="latent").images
-        image = upscaler(prompt=upscale_prompt, negative_prompt=upscale_neg, image=low_res_latents, num_inference_steps=upscale_steps, guidance_scale=upscale_scale, generator=generator).images[0]
+        image = upscaler(Prompt, negative_prompt=negative_prompt, image=low_res_latents, num_inference_steps=5, guidance_scale=0, generator=generator).images[0]
     else:
         image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
     return image
@@ -28,11 +28,6 @@ gr.Interface(fn=genie, inputs=[gr.Textbox(label='What you want the AI to generat
                                gr.Slider(25, maximum=100, value=50, step=25), 
                                gr.Slider(minimum=1, step=1, maximum=9999999999999999, randomize=True), 
                                gr.Radio(["Yes", "No"], label='Upscale?'),
-                               gr.Textbox(label='Upscaler Prompt: Optional'),
-                               gr.Textbox(label='Upscaler Negative Prompt: Both Optional And Experimental'),
-                               gr.Slider(minimum=0, maximum=15, value=0, step=1, label='Upscale Guidance Scale'),
-                               gr.Slider(minimum=5, maximum=25, value=5, step=5, label='Upscaler Iterations')
-             
                               ],
              outputs=gr.Image(label='Generated Image'), 
              title="PhotoReal V2 with SD x2 Upscaler - GPU",