Update handler.py

handler.py

@@ -5,7 +5,7 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from huggingface_inference_toolkit.logging import logger
 
-
+model_dir = 'Ephraimmm/PIDGIN_gemma-3'
 class EndpointHandler:
     """
     Custom handler for Hugging Face Inference Endpoints.
@@ -25,7 +25,7 @@ class EndpointHandler:
             self.model = AutoModelForCausalLM.from_pretrained(model_dir)
         except Exception as e:
             # fallback: try loading from hub name
-            hf_model_id = os.getenv("MODEL_ID") or "Ephraimmm/PIDGIN_gemma-3"
+            hf_model_id = "Ephraimmm/PIDGIN_gemma-3"
             logger.warn(f"Loading model from hub id {hf_model_id} because fallback. Exception: {e}")
             self.tokenizer = AutoTokenizer.from_pretrained(hf_model_id, use_fast=False)
             self.model = AutoModelForCausalLM.from_pretrained(hf_model_id)