narasimha01tm commited on
Commit
cac174e
Β·
verified Β·
1 Parent(s): 2117f6d

Upload server.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. server.py +5 -0
server.py CHANGED
@@ -23,6 +23,11 @@ import time
23
  import base64
24
  import logging
25
  import json
 
 
 
 
 
26
  from pathlib import Path
27
  from typing import Optional
28
 
 
23
  import base64
24
  import logging
25
  import json
26
+
27
+ # ── Suppress GPU/CUDA warnings on CPU Space ────────────────
28
+ os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
29
+ os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
30
+
31
  from pathlib import Path
32
  from typing import Optional
33