File size: 284 Bytes
0533273
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
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)