notabaka commited on
Commit
c208ca1
·
1 Parent(s): 47fd8f9
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ import tempfile
7
  import textract
8
  import docx2txt
9
  import pdfplumber
 
10
 
11
  def last_token_pool(last_hidden_states: Tensor,
12
  attention_mask: Tensor) -> Tensor:
 
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: