JoPmt commited on
Commit
36ff355
·
1 Parent(s): 6688fb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,7 +18,7 @@ warp.scheduler = DDPMScheduler.from_config(warp.scheduler.config)
18
  warp = warp.to("cpu")
19
  generator = torch.Generator(device="cpu").manual_seed(random.randint(1, 4876364))
20
 
21
- def plex(cook, one, two, three):
22
  ###goof = load_image(img).resize((512, 512))
23
  negative_prompt = "lowres,text,bad quality,low quality,jpeg artifacts,ugly,bad hands,bad face,blurry,bad eyes,watermark,signature"
24
  warp_out = warp_prior(prompt=cook, height=512,width=512,negative_prompt=negative_prompt,guidance_scale=4.0, num_inference_steps=5,generator=generator)
@@ -26,6 +26,6 @@ def plex(cook, one, two, three):
26
  imas = warp(**warp_out, height=512, width=512, num_inference_steps=5, prompt=cook,negative_prompt=primpt,guidance_scale=0.0,output_type="pil",generator=generator).images[0]
27
  return imas
28
 
29
- iface = gr.Interface(fn=plex,inputs=[gr.Textbox(label="prompt"), gr.Slider(label="Text Guide",minimum=0.01,step=0.01,maximum=0.99,value=0.5), gr.Slider(label="Your Image Guide",minimum=0.01,step=0.01,maximum=0.99,value=0.5),gr.Slider(label="Generated Image Guide",minimum=0.01,step=0.01,maximum=0.99,value=0.3)], outputs=gr.Image(), title="Txt2Img Wrstchn Interp", description="Txt2Img Wrstchn Interp")
30
  iface.queue(max_size=1)
31
  iface.launch(max_threads=1)
 
18
  warp = warp.to("cpu")
19
  generator = torch.Generator(device="cpu").manual_seed(random.randint(1, 4876364))
20
 
21
+ def plex(cook, one, two):
22
  ###goof = load_image(img).resize((512, 512))
23
  negative_prompt = "lowres,text,bad quality,low quality,jpeg artifacts,ugly,bad hands,bad face,blurry,bad eyes,watermark,signature"
24
  warp_out = warp_prior(prompt=cook, height=512,width=512,negative_prompt=negative_prompt,guidance_scale=4.0, num_inference_steps=5,generator=generator)
 
26
  imas = warp(**warp_out, height=512, width=512, num_inference_steps=5, prompt=cook,negative_prompt=primpt,guidance_scale=0.0,output_type="pil",generator=generator).images[0]
27
  return imas
28
 
29
+ iface = gr.Interface(fn=plex,inputs=[gr.Textbox(label="prompt"), gr.Slider(label="Inference steps",minimum=1,step=1,maximum=10,value=5), gr.Slider(label="Prior guidance scale",minimum=4.1,step=0.1,maximum=19.9,value=4.1)], outputs=gr.Image(), title="Txt2Img Wrstchn SD", description="Txt2Img Wrstchn SD")
30
  iface.queue(max_size=1)
31
  iface.launch(max_threads=1)