Spaces:
Runtime error
Runtime error
| 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() | |