up

app.py

@@ -307,10 +307,10 @@ def upscale_image(img, rows, cols, seed, prompt, negative_prompt, xformers, cpu_
     return final_img
 
 
-def upscale(mode, image, prompt):
-    print("upscale", mode, image, prompt)   
+def upscale( image, prompt):
+    print("upscale", image, prompt)   
     # return upscale1(image, prompt)
-    return upscale_image(image,rows=3,cols=3,seed=-1,prompt=prompt,negative_prompt="jpeg artifacts, lowres, bad quality, watermark",xformers=True,cpu_offload=True,attention_slicing=True,iterations=10)
+    return upscale_image(image,rows=4,cols=4,seed=-1,prompt=prompt,negative_prompt="jpeg artifacts, lowres, bad quality, watermark",xformers=True,cpu_offload=True,attention_slicing=True,iterations=10)
 
 modes = {
     '1': '1',
@@ -359,8 +359,7 @@ with gr.Blocks() as app:
             gr.Button("Upscale").click(
                 upscale, 
                 inputs=[
-                    gr.Radio(label="Mode", choices=list(modes.values())[:4], value=modes['upscale4x']),
-                    gr.Image(type="pil"),
+                    gr.Image(),
                     gr.Textbox(label="prompt",value="empty room")
                     ], 
                 outputs=gr.Image())