Dyna-99 commited on
Commit
dbae4d1
·
verified ·
1 Parent(s): 7a790c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -6,6 +6,12 @@ import re
6
  from flask import Flask, request, jsonify
7
  from sentence_transformers import SentenceTransformer
8
  from sklearn.metrics.pairwise import cosine_similarity
 
 
 
 
 
 
9
 
10
 
11
  class ImprovedBPJSChatbot:
 
6
  from flask import Flask, request, jsonify
7
  from sentence_transformers import SentenceTransformer
8
  from sklearn.metrics.pairwise import cosine_similarity
9
+ import os
10
+
11
+ os.environ['HF_HOME'] = '/tmp/huggingface'
12
+ os.environ['TRANSFORMERS_CACHE'] = '/tmp/huggingface/transformers'
13
+ os.environ['HF_DATASETS_CACHE'] = '/tmp/huggingface/datasets'
14
+ os.environ['HF_METRICS_CACHE'] = '/tmp/huggingface/metrics'
15
 
16
 
17
  class ImprovedBPJSChatbot: