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):
|