notabaka commited on
Commit
6ed8967
1 Parent(s): f52a963
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -8,6 +8,7 @@ import textract
8
  import docx2txt
9
  import pdfplumber
10
  import io
 
11
 
12
  def last_token_pool(last_hidden_states: Tensor,
13
  attention_mask: Tensor) -> Tensor:
 
8
  import docx2txt
9
  import pdfplumber
10
  import io
11
+ import os
12
 
13
  def last_token_pool(last_hidden_states: Tensor,
14
  attention_mask: Tensor) -> Tensor: