tensorgirl commited on
Commit
9cb52f4
·
verified ·
1 Parent(s): 6aebca7

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -0
main.py CHANGED
@@ -3,6 +3,9 @@ from app import predict
3
  import os
4
  from huggingface_hub import login
5
  from pydantic import BaseModel
 
 
 
6
 
7
  os.environ['HF_HOME'] = '/hug/cache/'
8
  os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
 
3
  import os
4
  from huggingface_hub import login
5
  from pydantic import BaseModel
6
+ import sys
7
+
8
+ sys.setrecursionlimit(10**6)
9
 
10
  os.environ['HF_HOME'] = '/hug/cache/'
11
  os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'