MrSimple01 commited on
Commit
1437fcb
·
verified ·
1 Parent(s): f3c5ac6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -45,6 +45,9 @@ class TextSegmentAnalysis(BaseModel):
45
  segments: List[Segment] = Field(description="List of text segments with analysis")
46
 
47
 
 
 
 
48
  tokenizer = AutoTokenizer.from_pretrained(TOKENIZER_MODEL)
49
  sentence_model = SentenceTransformer(SENTENCE_TRANSFORMER_MODEL)
50
 
 
45
  segments: List[Segment] = Field(description="List of text segments with analysis")
46
 
47
 
48
+ hf_token = os.environ.get('HF_TOKEN', None)
49
+ login(token=hf_token)
50
+
51
  tokenizer = AutoTokenizer.from_pretrained(TOKENIZER_MODEL)
52
  sentence_model = SentenceTransformer(SENTENCE_TRANSFORMER_MODEL)
53