Update diffusion.py

diffusion.py

@@ -10,6 +10,8 @@ def im_2_bytes(img_path):
     return (im)
 def gen_image(prompt,model_name="dreamlike-art/dreamlike-photoreal-2.0",count = 1):
     proc=gr.load(f"models/{model_name}")
+    output=proc(f'{prompt}')
+    '''
     t=0
     output_list=[]
     while t<int(count):
@@ -18,5 +20,6 @@ def gen_image(prompt,model_name="dreamlike-art/dreamlike-photoreal-2.0",count =
         inputs=f"{inputs} "
         t+=1
     #return(output_list)
+    '''
     return(im_2_bytes(output))