Spaces:

Arkm20
/

image_api

Sleeping

Arkm20 commited on Apr 15

Commit

0d3710f

verified ·

1 Parent(s): a9fed93

Update app/main.py

Files changed (1) hide show

app/main.py CHANGED Viewed

@@ -27,7 +27,7 @@ def root():
 @app.get("/gen")
 def generate_image(
     prompt: str = Query(..., description="Prompt for image generation"),
-    basemodel: str = "black-forest-labs/FLUX.1-schnell",
     width: int = 1280,
     height: int = 768,
     scales: int = 8,
@@ -35,7 +35,7 @@ def generate_image(
     seed: int = -1,
     upscale_factor: str = "2",
     process_upscale: bool = False,
-    lora_model: str = "XLabs-AI/flux-RealismLora",
     process_lora: bool = False
 ):
     try:

 @app.get("/gen")
 def generate_image(
     prompt: str = Query(..., description="Prompt for image generation"),
+    basemodel: str = "black-forest-labs/FLUX.1-DEV",
     width: int = 1280,
     height: int = 768,
     scales: int = 8,
     seed: int = -1,
     upscale_factor: str = "2",
     process_upscale: bool = False,
+    lora_model: str = "enhanceaiteam/Flux-uncensored",
     process_lora: bool = False
 ):
     try: