Spaces:

Omnibus-archive
/

Functions

No application file

Omnibus commited on Dec 29, 2023

Commit

25bbd1e

1 Parent(s): 0533273

Update diffusion.py

Files changed (1) hide show

diffusion.py CHANGED Viewed

@@ -1,6 +1,6 @@
 import gradio as gr
-def gen_image(prompt,model_name,count = 1):
     proc=gr.load(f"models/{model_name}")
     t=0
     output_list=[]

 import gradio as gr
+def gen_image(prompt,model_name="dreamlike-art/dreamlike-photoreal-2.0",count = 1):
     proc=gr.load(f"models/{model_name}")
     t=0
     output_list=[]