initlaize fastapi endpoint and dockerfile

Dockerfile

@@ -0,0 +1,14 @@
+# Use official Python runtime base image
+FROM python:3.10.9
+
+# Copy project files
+COPY . .
+
+# Set working directory
+WORKDIR /
+
+# Install dependencies listed in requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+# Command to run FastAPI app on port 7860
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,24 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from gradio_client import Client
+
+app = FastAPI()
+
+# Create Gradio client instance once
+client = Client("https://kby-ai-facerecognition.hf.space/--replicas/ij0t7/")
+
+# Pydantic model for input URLs
+class ImagesInput(BaseModel):
+    image1: str
+    image2: str
+
+@app.post("/face-recognition/")
+async def face_recognition(data: ImagesInput):
+    # Call the Gradio client predict with the URLs from request body
+    result = client.predict(
+        data.image1,  # parameter_8 image
+        data.image2,  # parameter_11 image
+        fn_index=2
+    )
+    # Return the prediction result as JSON
+    return {"result": result}

requirements.txt

@@ -0,0 +1,3 @@
+fastapi==0.99.1
+uvicorn[standard]==0.22.0
+gradio-client==0.5.3