PreranTej commited on
Commit
4bfcca6
·
verified ·
1 Parent(s): 358e141

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ ID_TO_BIAS = {
35
  # ── Load model once at startup ──
36
  print("Loading model...")
37
  MODEL_ID = "PreranTej/roberta-cognitive-bias-multitask"
38
- tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
39
  model = RobertaMultiTask.from_pretrained(MODEL_ID, num_labels=9)
40
  model.eval()
41
  print("Model loaded successfully")
 
35
  # ── Load model once at startup ──
36
  print("Loading model...")
37
  MODEL_ID = "PreranTej/roberta-cognitive-bias-multitask"
38
+ tokenizer = AutoTokenizer.from_pretrained("roberta-base")
39
  model = RobertaMultiTask.from_pretrained(MODEL_ID, num_labels=9)
40
  model.eval()
41
  print("Model loaded successfully")