Upload folder using huggingface_hub

handler.py

@@ -3,13 +3,11 @@ import torch
 import os
 from transformers import AutoTokenizer
 # Import your custom model class
-import sys
-# Add the local directory containing modelling_trm.py to the Python path
-sys.path.insert(0, ".") # Assuming the handler will be in the root of the repo
+# Assume the inference environment handles adding the repository root to the Python path
 from modelling_trm import TRM, TRMConfig
-sys.path.pop(0) # Remove the path after import
 
-class EndpointHandler: # Renamed class
+
+class EndpointHandler:
     def __init__(self):
         self.model = None
         self.tokenizer = None