lkjjj26 commited on
Commit
1cd3295
ยท
1 Parent(s): 1a3570b

update os env variable

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -15,11 +15,8 @@ warnings.filterwarnings('ignore')
15
  # Load environment variables from .env file
16
  load_dotenv()
17
 
18
- cache_dir = "./transformers_cache"
19
- os.makedirs(cache_dir, exist_ok=True) # ๋””๋ ‰ํ† ๋ฆฌ ์ƒ์„ฑ
20
- os.environ["TRANSFORMERS_CACHE"] = cache_dir # Hugging Face ์บ์‹œ ๊ฒฝ๋กœ ์„ค์ •
21
-
22
- print("Hugging Face Cache Path:", os.getenv("TRANSFORMERS_CACHE"))
23
 
24
  class PDBSearchAssistant:
25
  def __init__(self, model_name="google/flan-t5-large"):
 
15
  # Load environment variables from .env file
16
  load_dotenv()
17
 
18
+ os.environ["TRANSFORMERS_CACHE"] = "./transformers_cache"
19
+ os.makedirs("./transformers_cache", exist_ok=True)
 
 
 
20
 
21
  class PDBSearchAssistant:
22
  def __init__(self, model_name="google/flan-t5-large"):