diffusion.py

import gradio as gr

def gen_image(prompt,model_name,count = 1):
    proc=gr.load(f"models/{model_name}")
    t=0
    output_list=[]
    while t<int(count):
        output=proc(prompt)
        output_list.append(output)
        inputs=f"{inputs} "
        t+=1
    return(output_list)