Remove key print

nubebytes · Sep 27, 2022 · 29120b4 · 29120b4
1 parent 785d8cf
commit 29120b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lib/fits/hf/client/hf.py b/src/lib/fits/hf/client/hf.py
@@ -9,7 +9,7 @@
 YODA_FITS_MODEL = str(config("YODA_FITS_MODEL"))
 
 
-def _query(payload: FITSInferPayload, url: str):
+def query(payload: FITSInferPayload, url: str):
     request_url = f"{HF_BASE_URL}/{url}"
     headers = {"Authorization": f"Bearer {HF_API_KEY}"}
     response = requests.post(request_url, headers=headers, json=payload)