Rifky commited on
Commit
d446b84
1 Parent(s): b549dc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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):