Spaces:
Running
Running
import gradio as gr | |
class SendIt: | |
def __init__(self): | |
models =[ | |
"", | |
"CompVis/stable-diffusion-v1-4", | |
"runwayml/stable-diffusion-v1-5", | |
"prompthero/openjourney", | |
"stabilityai/stable-diffusion-2-1", | |
"stabilityai/stable-diffusion-2-1-base", | |
"SG161222/Realistic_Vision_V1.4", | |
"Linaqruf/anything-v3.0", | |
"eimiss/EimisAnimeDiffusion_1.0v", | |
"nitrosocke/Nitro-Diffusion", | |
"wavymulder/portraitplus", | |
"22h/vintedois-diffusion-v0-1", | |
"dreamlike-art/dreamlike-photoreal-2.0", | |
"dreamlike-art/dreamlike-diffusion-1.0", | |
"wavymulder/Analog-Diffusion", | |
"nitrosocke/redshift-diffusion", | |
"claudfuen/photorealistic-fuen-v1", | |
"prompthero/openjourney-v2", | |
"johnslegers/epic-diffusion", | |
"nitrosocke/Arcane-Diffusion", | |
"darkstorm2150/Protogen_x5.8_Official_Release", | |
] | |
self.models2=[ | |
gr.Interface.load(f"models/{models[1]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[2]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[3]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[4]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[5]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[6]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[7]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[8]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[9]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[10]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[11]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[12]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[13]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[14]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[15]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[16]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[17]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[18]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[19]}",live=True,preprocess=True), | |
gr.Interface.load(f"models/{models[20]}",live=True,preprocess=True), | |
] | |
do_it=SendIt() | |
def send_it(inputs,model_choice=5, count = 1): | |
t=0 | |
output_list=[] | |
while t<int(count): | |
proc1=do_it.models2[int(model_choice)] | |
output1=proc1(inputs) | |
output_list.append(output1) | |
inputs=f"{inputs} " | |
t+=1 | |
return(output_list) | |