MODIFIED: default model for text2image due to busy server

AquibPy · May 9, 2024 · bc95450 · bc95450
1 parent a97495a
commit bc95450
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api.py b/api.py
@@ -412,7 +412,7 @@ async def ats(resume_pdf: UploadFile = File(...),job_description: str = Form(...
 
             4. Stable_Diffusion_v2 - The latest version of Stable Diffusion, with improved performance and quality compared to the base version.
         """)
-def generate_image(prompt: str = Form("Astronaut riding a horse"), model: str = Form("DreamShaper_v7")):
+def generate_image(prompt: str = Form("Astronaut riding a horse"), model: str = Form("Stable_Diffusion_base")):
     try:
         if model in diffusion_models:
             def query(payload):