Aekanun commited on
Commit
7791806
·
1 Parent(s): 3f98862
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -11,6 +11,13 @@ try:
11
  except Exception as e:
12
  print(f"Failed to install packages: {e}")
13
 
 
 
 
 
 
 
 
14
  import warnings
15
  import torch
16
 
 
11
  except Exception as e:
12
  print(f"Failed to install packages: {e}")
13
 
14
+
15
+ ###
16
+ # เพิ่มบรรทัดนี้ที่ต้นโค้ด ก่อน import torch
17
+ import os
18
+ os.environ['NVIDIA_VISIBLE_DEVICES'] = ''
19
+ ###
20
+
21
  import warnings
22
  import torch
23