Update checks/endpoint_check.py

checks/endpoint_check.py

@@ -3,7 +3,7 @@ import sys
 
 import requests
 
-def is_huggingface_public_endpoint(endpoint: str):
+def is_public_endpoint(endpoint: str):
     try:
         # No Authorization header required for public models
         response = requests.get(endpoint)
@@ -23,15 +23,6 @@ def is_huggingface_public_endpoint(endpoint: str):
         print(f"Request failed: {e}")
         return False
 
-# Example usage:
-# Replace with your public Hugging Face model's endpoint URL
-endpoint = "https://api-inference.huggingface.co/models/gpt2"  # Example public model URL
-
-if is_huggingface_public_endpoint(endpoint):
-    print("This is a valid Hugging Face Inference Endpoint.")
-else:
-    print("This is NOT a Hugging Face Inference Endpoint.")
-
 
 
 def is_huggingface_endpoint(endpoint: str):