add: fastapi

app.py

@@ -9,6 +9,8 @@ import spaces
 import cv2
 import base64
 from io import BytesIO
+import uvicorn
+from fastapi import app
 from diffusers import StableDiffusionPipeline, StableDiffusionInpaintPipeline
 
 model_id = "stabilityai/stable-diffusion-2-inpainting"
@@ -95,6 +97,10 @@ def clothing_try_on_base64(input_image_base64, mask_image_base64):
     return image_to_base64(output_image)
 
 
+def run_fastapi():
+    uvicorn.run(app, host="0.0.0.0", port=7860)
+
+
 def launch_interface_base64():
     with gr.Blocks() as interface:
         with gr.Row():
@@ -105,6 +111,7 @@ def launch_interface_base64():
         submit = gr.Button("Apply")
 
         submit.click(fn=clothing_try_on_base64, inputs=[inputImage, maskImage], outputs=[outputOne])
+        submit.click(fn=run_fastapi, inputs=[], outputs=[])
 
         interface.launch(debug=True)
 

fastapi.py

@@ -0,0 +1,17 @@
+"""
+Created By: ishwor subedi
+Date: 2024-10-04
+"""
+from CTO_TCP_V1.app import clothing_try_on_base64
+from fastapi import FastAPI
+
+app = FastAPI(title="CTO TCP V1")
+
+
+@app.post("/clothingTryOn")
+async def clothing_try_on(image: str, mask: str):
+    result = clothing_try_on_base64(image, mask)
+
+    return {"result": result}
+
+