Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -53,12 +53,15 @@ models =[
|
|
53 |
"digiplay/PotoPhotoRealism_v1",
|
54 |
]
|
55 |
|
|
|
|
|
|
|
56 |
def plex(prompt,modil):
|
57 |
pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained(""+modil+"", torch_dtype=torch.float32))
|
58 |
pipe = accelerator.prepare(pipe.to("cpu"))
|
59 |
image = pipe(prompt=prompt, num_inference_steps=10).images[0]
|
60 |
return image
|
61 |
|
62 |
-
iface = gr.Interface(fn=plex,outputs=gr.Image(label="Generated Output Image"),inputs=[gr.Dropdown(choices=models, type="value", value=models[
|
63 |
iface.queue(max_size=1)
|
64 |
iface.launch(max_threads=1)
|
|
|
53 |
"digiplay/PotoPhotoRealism_v1",
|
54 |
]
|
55 |
|
56 |
+
bor = len(models)
|
57 |
+
current = random.randint(1, bor)
|
58 |
+
|
59 |
def plex(prompt,modil):
|
60 |
pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained(""+modil+"", torch_dtype=torch.float32))
|
61 |
pipe = accelerator.prepare(pipe.to("cpu"))
|
62 |
image = pipe(prompt=prompt, num_inference_steps=10).images[0]
|
63 |
return image
|
64 |
|
65 |
+
iface = gr.Interface(fn=plex,outputs=gr.Image(label="Generated Output Image"),inputs=[gr.Dropdown(choices=models, type="value", value=models[current]), gr.Textbox(label="Prompt"), gr.Textbox(label="negative_prompt", value="low quality, bad quality")])], title="AutoPipelineForText2Image_SD_Multi",description="AutoPipelineForText2Image_SD_Multi")
|
66 |
iface.queue(max_size=1)
|
67 |
iface.launch(max_threads=1)
|