++

Dockerfile

@@ -10,4 +10,4 @@ COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY --chown=user . /app
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--debug"]

routes/textToImage.py

@@ -6,18 +6,18 @@ import io
 router = APIRouter()
 
 @router.post("/v1/images/generations", tags=["Text to Image"])
-async def text_to_image(body: TextToImageRequest):
-    client = InferenceClient(model=body.model)
+async def text_to_image(t2i_body: TextToImageRequest):
+    client = InferenceClient(model=t2i_body.model)
     res = client.text_to_image(
-        prompt=body.prompt,
-        negative_prompt=body.negative_prompt,
-        height=body.height,
-        width=body.width,
-        num_inference_steps=body.num_inference_steps,
-        guidance_scale=body.guidance_scale,
-        scheduler=body.scheduler,
-       # target_size=body.target_size,
-        seed=body.seed
+        prompt=t2i_body.prompt,
+        negative_prompt=t2i_body.negative_prompt,
+        height=t2i_body.height,
+        width=t2i_body.width,
+        num_inference_steps=t2i_body.num_inference_steps,
+        guidance_scale=t2i_body.guidance_scale,
+        scheduler=t2i_body.scheduler,
+       # target_size=t2i_body.target_size,
+        seed=t2i_body.seed
     )
     img_byte_arr = io.BytesIO()
     res.save(img_byte_arr, format="PNG")