Update pages/type_text.py
Browse files- pages/type_text.py +1 -1
pages/type_text.py
CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
|
|
2 |
import pandas as pd
|
3 |
from io import StringIO
|
4 |
import json
|
5 |
-
|
6 |
from transformers import pipeline # AutoTokenizer, AutoModelForCausalLM, AutoModelForTokenClassification
|
7 |
from sentence_transformers import SentenceTransformer, util
|
8 |
|
|
|
2 |
import pandas as pd
|
3 |
from io import StringIO
|
4 |
import json
|
5 |
+
import torch
|
6 |
from transformers import pipeline # AutoTokenizer, AutoModelForCausalLM, AutoModelForTokenClassification
|
7 |
from sentence_transformers import SentenceTransformer, util
|
8 |
|