Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
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="
|
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)
|