Update checks/endpoint_check.py

checks/endpoint_check.py

@@ -1,6 +1,39 @@
 import requests
 import sys
 
+import requests
+
+def is_huggingface_public_endpoint(endpoint: str):
+    try:
+        # No Authorization header required for public models
+        response = requests.get(endpoint)
+
+        # Check if the response status code is 200 and it returns inference data
+        if response.status_code == 200:
+            # Public models will return inference data without needing an API key
+            if "model" in response.json() or "error" in response.json():
+                return True
+            else:
+                print("The response does not contain inference-related data.")
+                return False
+        else:
+            print(f"Failed to reach the endpoint, status code: {response.status_code}")
+            return False
+    except requests.exceptions.RequestException as e:
+        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):
     try:
         headers = {"Authorization": "Bearer YOUR_HUGGINGFACE_API_KEY"}