Home / app.py
Alashazam's picture
Update app.py
8c4e202
import gradio
class Model:
def __init__(self, name, path="", prefix=""):
self.name = name
self.path = path
self.prefix = prefix
models = [
Model("Marvel","models/ItsJayQz/Marvel_WhatIf_Diffusion", "whatif style"),
Model("Portrait plus", "models/wavymulder/portraitplus", "portrait+ style"),
Model("CF25", "models/gsdf/Counterfeit-V2.5", "anime style"),
Model("vintedois", "models/22h/vintedois-diffusion-v0-1", "vintedois style"),
Model("dreamlike", "models/dreamlike-art/dreamlike-diffusion-1.0","dreamlike style"),
Model("GTA5","models/ItsJayQz/GTA5_Artwork_Diffusion", "GTA5 style")
]
model1=[]
model2=[]
model3=[]
for i in range(len(models)):
model3.append(models[i].name)
model1.append(gradio.Interface.load(models[i].path))
model2.append(models[i].prefix)
def process1(prompt, modelSelected):
if (modelSelected==''):
modelSelected = "Marvel"
model_idx=model3.index(modelSelected)
prompt+=", in "+model2[model_idx]
image_return = model1[model_idx](prompt)
return image_return
sandbox = gradio.Interface(fn=process1,
inputs=[gradio.Textbox(label="Enter Prompt:"), gradio.Dropdown(model3)],
outputs=[gradio.Image(label="Produced Image")],
title='Text to Image',
examples=[
["Viggo Mortensen Gryffindor wizard portrait, Hogwart University, castle tower background", "Portrait plus"],
["1girl pirate, left patch, detailed face, black hat, big sailing boat, ocean in background", "CF25"],
["Portrait close up, Elvis Presley, concert hall in the background", "GTA5"],
["Marvel Blackwidow portrait close up. building city background", "Marvel"],
["close up portrait Benedict Cumberbatch wizard of black magic, robe with hood, Hogwart University, castle tower background, oil painting on canvas", "vintedois"],
["A white rabbit wizard, Hogwart University, Castle in the background", "dreamlike"]
])
sandbox.queue(concurrency_count=20).launch()