a0ms1n commited on
Commit
0920d0b
·
verified ·
1 Parent(s): ea95da5

Fix crashes

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -3,6 +3,15 @@ import Evaluate1
3
  import Evaluate2
4
  import re
5
 
 
 
 
 
 
 
 
 
 
6
  def load_cpp_file(file):
7
  try:
8
  text = file.name
 
3
  import Evaluate2
4
  import re
5
 
6
+ import os
7
+ import torch
8
+
9
+ # CRITICAL: Redirect cache to temporary storage
10
+ os.environ['TORCH_HOME'] = '/tmp/torch_cache'
11
+ os.environ['HUB_DIR'] = '/tmp/torch_hub'
12
+ os.environ['TMPDIR'] = '/tmp'
13
+ torch.hub.set_dir('/tmp/torch_hub')
14
+
15
  def load_cpp_file(file):
16
  try:
17
  text = file.name