Hasitha16 commited on
Commit
c06888f
·
verified ·
1 Parent(s): b5ad127

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +3 -3
model.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  from typing import List, Optional
2
  from pydantic import BaseModel
3
  from transformers import pipeline
@@ -9,9 +11,7 @@ from sklearn.cluster import KMeans
9
  from nltk.tokenize import sent_tokenize
10
  from sklearn.metrics.pairwise import cosine_similarity
11
  import numpy as np
12
- import os
13
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
14
- # 📄 Load HuggingFace Pipelines
15
  summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
16
  sentiment_analyzer = pipeline("sentiment-analysis")
17
 
 
1
+ import os
2
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf-cache"
3
  from typing import List, Optional
4
  from pydantic import BaseModel
5
  from transformers import pipeline
 
11
  from nltk.tokenize import sent_tokenize
12
  from sklearn.metrics.pairwise import cosine_similarity
13
  import numpy as np
14
+
 
 
15
  summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
16
  sentiment_analyzer = pipeline("sentiment-analysis")
17