change to short model

README.md

@@ -9,7 +9,7 @@ app_port: 7860
 
 # AI Text Detector API
 
-This is an API service for detecting AI-generated text. It uses the `yuchuantian/AIGC_detector_env3` model (quantized for CPU efficiency) and FastText for language validation.
+This is an API service for detecting AI-generated text. It uses the `yuchuantian/AIGC_detector_env3short` model (quantized for CPU efficiency) and FastText for language validation.
 
 ## Features
 

download_model.py

@@ -4,7 +4,7 @@ import requests
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
 # Configuration
-MODEL_TAG = "yuchuantian/AIGC_detector_env3"
+MODEL_TAG = "yuchuantian/AIGC_detector_env3short"
 MODELS_DIR = os.getenv("MODELS_DIR", "saved_models")
 QUANTIZED_CPU_FILENAME = "aigc_detector_int8.pt"
 FASTTEXT_MODEL_URL = "https://dl.fbaipublicfiles.com/fasttext/supervised-models/lid.176.ftz"

src/config.py

@@ -18,7 +18,7 @@ class Config:
     FASTTEXT_FILENAME = "lid.176.ftz"
 
     # AI Detector Model
-    # AI_DETECTOR_MODEL_NAME = "yuchuantian/AIGC_detector_env3" # Deprecated in favor of local quantized
+    # AI_DETECTOR_MODEL_NAME = "yuchuantian/AIGC_detector_env3short" # Deprecated in favor of local quantized
     AI_DETECTOR_FILENAME = "aigc_detector_int8.pt"
 
     @staticmethod

src/models/ai_detector_model.py

@@ -37,7 +37,7 @@ class AIDetectorModel:
             logger.warning(f"Failed to load quantized model: {e}. Attempting fallback...")
 
         # 2. Fallback to base Hugging Face model
-        fallback_tag = "yuchuantian/AIGC_detector_env3"
+        fallback_tag = "yuchuantian/AIGC_detector_env3short"
         logger.info(f"Loading fallback model: {fallback_tag}...")
         
         try: