Add HF_TOKEN support for private model access

app.py

@@ -32,8 +32,9 @@ import gradio as gr
 
 # ========== CONFIGURATION ==========
 
-# Hugging Face model ID
+# Hugging Face model ID and token
 HF_MODEL_ID = os.getenv("HF_MODEL_ID", "Meet2304/convnextv2-cervical-cell-classification")
+HF_TOKEN = os.getenv("HF_TOKEN")
 
 # Class names
 CLASS_NAMES = [
@@ -63,11 +64,11 @@ print(f"Model ID: {HF_MODEL_ID}")
 print(f"Device: {DEVICE}")
 
 # Load image processor
-processor = AutoImageProcessor.from_pretrained(HF_MODEL_ID)
+processor = AutoImageProcessor.from_pretrained(HF_MODEL_ID, token=HF_TOKEN)
 print("✓ Processor loaded")
 
 # Load model
-model = ConvNextV2ForImageClassification.from_pretrained(HF_MODEL_ID)
+model = ConvNextV2ForImageClassification.from_pretrained(HF_MODEL_ID, token=HF_TOKEN)
 model = model.to(DEVICE)
 model.eval()
 print("✓ Model loaded and set to evaluation mode")