whtet commited on
Commit
76ecd6c
·
verified ·
1 Parent(s): b38bcba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -16,8 +16,7 @@ def kill_port(port):
16
  kill_port(7860) # Ensure Flask doesn't crash due to a used port
17
 
18
  # Define cache directory
19
- cache_dir = "/app/cache"
20
- os.environ["TRANSFORMERS_CACHE"] = cache_dir # Set cache to writable folder
21
 
22
  # Load Myanmarsar-GPT (1.42B params) from Hugging Face
23
  MODEL_NAME = "simbolo-ai/Myanmarsar-GPT"
 
16
  kill_port(7860) # Ensure Flask doesn't crash due to a used port
17
 
18
  # Define cache directory
19
+ os.environ["HF_HOME"] = "/app/cache"
 
20
 
21
  # Load Myanmarsar-GPT (1.42B params) from Hugging Face
22
  MODEL_NAME = "simbolo-ai/Myanmarsar-GPT"