Create diffusion.py

diffusion.py

@@ -0,0 +1,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)