Spaces:
Sleeping
Sleeping
Update checks/endpoint_check.py
Browse files- checks/endpoint_check.py +1 -1
checks/endpoint_check.py
CHANGED
@@ -21,7 +21,7 @@ def is_public_endpoint(endpoint: str):
|
|
21 |
print(f"Response JSON keys: {list(response_json.keys())[:5]}")
|
22 |
|
23 |
# Public models will return inference data without needing an API key
|
24 |
-
|
25 |
return True
|
26 |
else:
|
27 |
print("The response does not contain inference-related data.")
|
|
|
21 |
print(f"Response JSON keys: {list(response_json.keys())[:5]}")
|
22 |
|
23 |
# Public models will return inference data without needing an API key
|
24 |
+
if "model" in response_json or "error" in response_json:
|
25 |
return True
|
26 |
else:
|
27 |
print("The response does not contain inference-related data.")
|