Add api_name to Gradio endpoints for API access

app.py

@@ -32,8 +32,8 @@ import gradio as gr
 
 # ========== CONFIGURATION ==========
 
-# Model path (local files in the Space)
-MODEL_PATH = os.path.dirname(os.path.abspath(__file__))
+# Hugging Face model ID
+HF_MODEL_ID = os.getenv("HF_MODEL_ID", "Meet2304/convnextv2-cervical-cell-classification")
 
 # Class names
 CLASS_NAMES = [
@@ -58,16 +58,16 @@ DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 # ========== MODEL LOADING ==========
 
-print("Loading model from local files...")
-print(f"Model path: {MODEL_PATH}")
+print("Loading model from Hugging Face...")
+print(f"Model ID: {HF_MODEL_ID}")
 print(f"Device: {DEVICE}")
 
-# Load image processor from local files
-processor = AutoImageProcessor.from_pretrained(MODEL_PATH)
+# Load image processor
+processor = AutoImageProcessor.from_pretrained(HF_MODEL_ID)
 print("✓ Processor loaded")
 
-# Load model from local files
-model = ConvNextV2ForImageClassification.from_pretrained(MODEL_PATH)
+# Load model
+model = ConvNextV2ForImageClassification.from_pretrained(HF_MODEL_ID)
 model = model.to(DEVICE)
 model.eval()
 print("✓ Model loaded and set to evaluation mode")
@@ -300,7 +300,8 @@ with gr.Blocks(css=custom_css, title="Project Phoenix - Cervical Cancer Cell Cla
             predict_btn_basic.click(
                 fn=predict_basic,
                 inputs=input_image_basic,
-                outputs=output_label_basic
+                outputs=output_label_basic,
+                api_name="predict_basic"
             )
         
         # Tab 2: Prediction with Explainability
@@ -320,7 +321,8 @@ with gr.Blocks(css=custom_css, title="Project Phoenix - Cervical Cancer Cell Cla
             predict_btn_explain.click(
                 fn=predict_with_explainability,
                 inputs=input_image_explain,
-                outputs=[output_label_explain, output_gradcam, output_info]
+                outputs=[output_label_explain, output_gradcam, output_info],
+                api_name="predict_with_explainability"
             )
     
     # Footer