Spaces:
Sleeping
Sleeping
Commit
·
3efec51
1
Parent(s):
e02dc08
Removed faulty imports
Browse files
app.py
CHANGED
|
@@ -151,9 +151,6 @@ def search(query, similarity=False):
|
|
| 151 |
)
|
| 152 |
|
| 153 |
### Prediction ###
|
| 154 |
-
from flair.models import TextClassifier
|
| 155 |
-
from flair.data import Sentence
|
| 156 |
-
from flair.tokenization import SegtokTokenizer
|
| 157 |
from transformers import (
|
| 158 |
AutoTokenizer,
|
| 159 |
AutoModelForSeq2SeqLM,
|
|
@@ -162,11 +159,6 @@ def search(query, similarity=False):
|
|
| 162 |
)
|
| 163 |
from sentence_transformers import SentenceTransformer, CrossEncoder
|
| 164 |
from sentence_transformers.util import cos_sim, dot_score
|
| 165 |
-
from optimum.onnxruntime import (
|
| 166 |
-
ORTModelForSeq2SeqLM,
|
| 167 |
-
ORTModelForSequenceClassification,
|
| 168 |
-
)
|
| 169 |
-
from optimum.pipelines import pipeline as optimum_pipeline
|
| 170 |
|
| 171 |
# Load the classifiers
|
| 172 |
# classifier = TextClassifier.load(
|
|
|
|
| 151 |
)
|
| 152 |
|
| 153 |
### Prediction ###
|
|
|
|
|
|
|
|
|
|
| 154 |
from transformers import (
|
| 155 |
AutoTokenizer,
|
| 156 |
AutoModelForSeq2SeqLM,
|
|
|
|
| 159 |
)
|
| 160 |
from sentence_transformers import SentenceTransformer, CrossEncoder
|
| 161 |
from sentence_transformers.util import cos_sim, dot_score
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 162 |
|
| 163 |
# Load the classifiers
|
| 164 |
# classifier = TextClassifier.load(
|