updata handle trh

handler.py

@@ -532,8 +532,30 @@ def get_model_info():
         "device": str(model.device) if model else "Unknown"
     }
 
+# Hugging Face EndpointHandler class
+class EndpointHandler:
+    """Hugging Face endpoint handler class"""
+    
+    def __init__(self, model_dir):
+        """Initialize the endpoint handler"""
+        self.model_dir = model_dir
+        print(f"EndpointHandler initialized with model_dir: {model_dir}")
+    
+    def __call__(self, payload):
+        """Main endpoint function - delegates to query function"""
+        return query(payload)
+    
+    def health_check(self):
+        """Health check endpoint"""
+        return health_check()
+    
+    def get_model_info(self):
+        """Get model information"""
+        return get_model_info()
+
 # For backward compatibility and testing
 if __name__ == "__main__":
     print("Handler module loaded successfully!")
     print("This handler is now ready for Hugging Face endpoints.")
-    print("Use the 'query' function as the main endpoint.")
+    print("Use the 'query' function as the main endpoint.")
+    print("Or use EndpointHandler class for Hugging Face compatibility.")