Spaces:

X-iZhang
/

RadEval

Running

X-iZhang commited on Jul 11

Commit

5322748

verified ·

1 Parent(s): a388fcc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,11 +1,12 @@
 import gradio as gr
 import sys
 import os
 def setup_cpu_environment():
     os.environ['CUDA_VISIBLE_DEVICES'] = ''
-    torch.set_num_threads(4)  # 根据你的 CPU 核心数调整
     os.environ['TOKENIZERS_PARALLELISM'] = 'false'

 import gradio as gr
 import sys
 import os
+import torch
 def setup_cpu_environment():
     os.environ['CUDA_VISIBLE_DEVICES'] = ''
+    torch.set_num_threads(4)
     os.environ['TOKENIZERS_PARALLELISM'] = 'false'