Don't show API unless debugging

h2oai · Apr 18, 2023 · 3548454 · 3548454
1 parent baaa5e4
commit 3548454
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/client_test.py b/client_test.py
@@ -6,14 +6,16 @@
 python generate.py  --base_model=h2oai/h2ogpt-oig-oasst1-256-6.9b --chat=False --stream_output=False
 """
 
+debug = False
 
 import time
 import os
 os.environ['HF_HUB_DISABLE_TELEMETRY'] = '1'
 from gradio_client import Client
 
 client = Client("http://localhost:7860")
-print(client.view_api(all_endpoints=True))
+if debug:
+    print(client.view_api(all_endpoints=True))
 
 instruction = "Who are you?"
 iinput = ''