apply python formatting

app/main.py

@@ -11,19 +11,15 @@ app = FastAPI(
     docs_url="/docs",
     redoc_url="/redoc",
     openapi_url="/openapi.json",
-    swagger_ui_parameters={
-        "tryItOutEnabled": True
-    }
+    swagger_ui_parameters={"tryItOutEnabled": True},
 )
 
 
 @app.get("/")
 def root():
-    return {
-        "message": "Tiny SD API is running"
-    }
+    return {"message": "Tiny SD API is running"}
+
 
-    
 @app.get("/generate")
 def generate(prompt: str):
     image = generate_image(prompt)
@@ -31,23 +27,12 @@ def generate(prompt: str):
     output_dir = Path("outputs")
     output_dir.mkdir(exist_ok=True)
 
-    safe_prompt = re.sub(
-        r'[\\/*?:"<>|]',
-        "_",
-        prompt
-    )[:40]
+    safe_prompt = re.sub(r'[\\/*?:"<>|]', "_", prompt)[:40]
 
-    filename = (
-        f"{safe_prompt}-"
-        f"{datetime.now().strftime('%Y%m%d_%H%M%S')}.png"
-    )
+    filename = f"{safe_prompt}-" f"{datetime.now().strftime('%Y%m%d_%H%M%S')}.png"
 
     path = output_dir / filename
 
     image.save(path)
 
-    return FileResponse(
-        path=path,
-        media_type="image/png",
-        filename=filename
-    )
+    return FileResponse(path=path, media_type="image/png", filename=filename)

app/model.py

@@ -8,7 +8,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
     cache_dir="./models",
     # local_files_only=True,
     torch_dtype=torch.float32,
-    safety_checker=None
+    safety_checker=None,
 )
 
 pipe.enable_attention_slicing()
@@ -19,11 +19,6 @@ pipe.to("cpu")
 @torch.inference_mode()
 def generate_image(prompt):
 
-    image = pipe(
-        prompt,
-        height=256,
-        width=256,
-        num_inference_steps=10
-    ).images[0]
+    image = pipe(prompt, height=256, width=256, num_inference_steps=10).images[0]
 
-    return image
+    return image