Spaces:

Phoenix1410
/

LEXGUARD_API

Sleeping

Phoenix1410 commited on Sep 23

Commit

c9cda31

verified ·

1 Parent(s): bd84662

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -21,6 +21,8 @@ from dotenv import load_dotenv
 import sys
 import io
 # Ensure UTF-8 stdout on Windows
 if sys.stdout and hasattr(sys.stdout, 'buffer'):
     sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8', errors='replace')

 import sys
 import io
+os.environ["USE_TF"] = "0"
+os.environ["USE_TORCH"] = "1"
 # Ensure UTF-8 stdout on Windows
 if sys.stdout and hasattr(sys.stdout, 'buffer'):
     sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8', errors='replace')