Merge branch 'main' of hf.co:spaces/khawir/cg-gen-ai

app/api/generate.py

@@ -145,6 +145,7 @@ def image_variations(model: Request, request: ImageVariations, db: Annotated[Ses
     image.resize((256, 256))
 
     if prompt:
+        prompt = f"best quality, high quality, {prompt}"
         crud.create_prompt(db=db, user_id=current_user.user_id, prompt=prompt)
     else:
         request.scale = 1.0

app/core/schemas.py

@@ -62,7 +62,7 @@ class TextImage(Generate):
 class ImageImage(Generate):
     prompt: str
     image: str
-    num_inference_steps: int = 30
+    num_inference_steps: int = 50
     guidance_scale: float = 7.5
     image_guidance_scale: float = 1.5
 
@@ -74,7 +74,7 @@ class BackgroundRemoval(BaseModel):
 class ImageVariations(Generate):
     image: str
     num_samples: int = 4
-    num_inference_steps: int = 30
+    num_inference_steps: int = 50
     prompt: str | None = ""
     scale: float = 0.5
     guidance_scale: float = 7.5