Joshua Lochner
commited on
Commit
·
fce5be3
1
Parent(s):
e513830
Update pipeline.py
Browse files- pipeline.py +0 -3
pipeline.py
CHANGED
|
@@ -287,9 +287,6 @@ def binary_search_above(transcript, start_index, end_index, time):
|
|
| 287 |
|
| 288 |
class PreTrainedPipeline():
|
| 289 |
def __init__(self, path: str):
|
| 290 |
-
# load the model
|
| 291 |
-
self.model = keras.models.load_model(os.path.join(path, "tf_model.h5"))
|
| 292 |
-
|
| 293 |
path2 = os.path.join(path, "model")
|
| 294 |
self.model2 = AutoModelForSequenceClassification.from_pretrained(path2)
|
| 295 |
self.tokenizer2 = AutoTokenizer.from_pretrained(path2)
|
|
|
|
| 287 |
|
| 288 |
class PreTrainedPipeline():
|
| 289 |
def __init__(self, path: str):
|
|
|
|
|
|
|
|
|
|
| 290 |
path2 = os.path.join(path, "model")
|
| 291 |
self.model2 = AutoModelForSequenceClassification.from_pretrained(path2)
|
| 292 |
self.tokenizer2 = AutoTokenizer.from_pretrained(path2)
|