Update app/main.py

app/main.py

@@ -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",
+    basemodel: str = "black-forest-labs/FLUX.1-DEV",
     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",
+    lora_model: str = "enhanceaiteam/Flux-uncensored",
     process_lora: bool = False
 ):
     try: