Update app.py
Browse files
app.py
CHANGED
|
@@ -23,7 +23,7 @@ def load_model():
|
|
| 23 |
model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint, num_labels=2)
|
| 24 |
base_model = SentenceTransformer(base_model_checkpoint)
|
| 25 |
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, fast=True)
|
| 26 |
-
data = load_dataset(data_checkpoint, split="train")
|
| 27 |
return model, base_model, tokenizer, data
|
| 28 |
|
| 29 |
def sigmoid(x):
|
|
|
|
| 23 |
model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint, num_labels=2)
|
| 24 |
base_model = SentenceTransformer(base_model_checkpoint)
|
| 25 |
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, fast=True)
|
| 26 |
+
data = load_dataset(data_checkpoint, split="train", download_mode='force_redownload')
|
| 27 |
return model, base_model, tokenizer, data
|
| 28 |
|
| 29 |
def sigmoid(x):
|