feat: added some required features

app.py

@@ -6,7 +6,9 @@ from slowapi.util import get_remote_address
 from fastapi.responses import JSONResponse
 from features.text_classifier.routes import router as text_classifier_router
 from features.nepali_text_classifier.routes import router as nepali_text_classifier_router
+from features.image_classifier.routes import router as image_classifier_router
 from config import ACCESS_RATE
+
 import requests
 limiter = Limiter(key_func=get_remote_address, default_limits=[ACCESS_RATE])
 
@@ -27,11 +29,12 @@ app.add_middleware(SlowAPIMiddleware)
 # Include your routes
 app.include_router(text_classifier_router, prefix="/text")
 app.include_router(nepali_text_classifier_router,prefix="/NP")
+app.include_router(image_classifier_router,prefix="/AI-image")
 @app.get("/")
 @limiter.limit(ACCESS_RATE)
 async def root(request: Request):
     return {
         "message": "API is working",
-        "endpoints": ["/text/analyse", "/text/upload", "/text/analyse-sentences", "/text/analyse-sentance-file"]
+        "endpoints": ["/text/analyse", "/text/upload", "/text/analyse-sentences", "/text/analyse-sentance-file","/NP/analyse","/NP/upload","/NP/analyse-sentences","/NP/file-sentences-analyse","/AI-image/analyse"]
     }
 

features/image_classifier/preprocess.py

@@ -0,0 +1,9 @@
+import cv2
+import numpy as np
+def image_preprocessing(img_path):
+    img =cv2.imread(img_path)
+    img = cv2.resize(img,(128,128))
+    img= cv2.cvtColor(img,cv2.COLOR_BayerGR2RGB)
+    img = img/255.0
+    img = np.expand_dims(img,axis=0)
+    return img

features/image_classifier/routes.py

@@ -0,0 +1,21 @@
+from slowapi import Limiter
+from config import ACCESS_RATE
+from fastapi import APIRouter, File, Request, Depends, HTTPException, UploadFile
+from fastapi.security import HTTPBearer
+from slowapi import Limiter
+from slowapi.util import get_remote_address
+router = APIRouter()
+limiter = Limiter(key_func=get_remote_address)
+security = HTTPBearer()
+
+
+@router.post("/analyse")
+@limiter.limit(ACCESS_RATE)
+async def analyse(request: Request, file:UploadFile,token: str = Depends(security)):
+    return {"filename": file}
+
+@router.get("/health")
+@limiter.limit(ACCESS_RATE)
+def health(request: Request):
+    return {"status": "ok"}
+

features/nepali_text_classifier/model_loader.py

@@ -8,7 +8,7 @@ from huggingface_hub import snapshot_download
 from transformers import AutoTokenizer, AutoModel
 
 # Configs
-REPO_ID = "Pujan-Dev/Nepali-AI-VS-HUMAN"
+REPO_ID = "can-org/Nepali-AI-VS-HUMAN"
 BASE_DIR = "./np_text_model"
 TOKENIZER_DIR = os.path.join(BASE_DIR, "classifier")  # <- update this to match your uploaded folder
 WEIGHTS_PATH = os.path.join(BASE_DIR, "model_95_acc.pth")  # <- change to match actual uploaded weight

features/text_classifier/model_loader.py

@@ -6,7 +6,7 @@ from huggingface_hub import snapshot_download
 import torch
 from dotenv import load_dotenv
 load_dotenv()
-REPO_ID = "Pujan-Dev/AI-Text-Detector"
+REPO_ID = "can-org/AI-Content-Checker"
 MODEL_DIR = "./models"
 TOKENIZER_DIR = os.path.join(MODEL_DIR, "model")
 WEIGHTS_PATH = os.path.join(MODEL_DIR, "model_weights.pth")

models/.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text