Upload 3 files

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.10-slim
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+WORKDIR $HOME/app
+COPY --chown=user . $HOME/app
+RUN pip install -r requirements.txt
+
+EXPOSE 7860
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,45 @@
+from fastapi import FastAPI,UploadFile,File,HTTPException
+from transformers import pipeline
+from facenet_pytorch import MTCNN
+from PIL import Image
+import io
+
+app=FastAPI(title="Deepfake")
+
+print("Face Detector Loading...")
+mtcnn=MTCNN(keep_all=False,device='cpu')
+
+print("Deepfake AI Loading...")
+pipe=pipeline("image-classification",model="dima806/deepfake_vs_real_image_detection")
+print("All Systems Loaded!")
+
+@app.post("/api/v1/predict/image")
+async def predict_image(file:UploadFile=File(...)):
+    if not file.content_type.startswith("image/"):
+        raise HTTPException(status_code=400,detail="File must be an image.")
+
+    try:
+        image_bytes=await file.read()
+        image=Image.open(io.BytesIO(image_bytes)).convert("RGB")
+        # check for human faces
+        boxes, _ =mtcnn.detect(image)
+        
+        if boxes is None:
+            return {
+                "status":"failed",
+                "message":"No human face detected in the image. Please upload a clear human portrait."
+            }
+
+        results=pipe(image)
+        formatted_results={res['label'].lower(): round(res['score'] * 100, 2) for res in results}
+        verdict=max(formatted_results,key=formatted_results.get)
+
+        return {
+            "status":"success 200",
+            "faces_detected":len(boxes),
+            "verdict":verdict,
+            "confidence":formatted_results[verdict]
+        }
+    
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))

requirements.txt

@@ -0,0 +1,10 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
+torch==2.2.2
+torchvision==0.17.2
+numpy<2.0.0
+transformers<4.44.0
+facenet-pytorch
+pillow
+fastapi
+uvicorn
+python-multipart