salexashenko commited on
Commit
7735bc8
β€’
1 Parent(s): a8207d3
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -348,12 +348,12 @@ def recreate_bm25_idx(
348
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
349
 
350
  safety_tokenizer = tokenizer = AutoTokenizer.from_pretrained(
351
- "sanagnos/galactica-6.7b-finetuned",
352
  use_auth_token=HF_TOKEN,
353
  )
354
  safety_model = model = (
355
  AutoModelForSeq2SeqLM.from_pretrained(
356
- "sanagnos/galactica-6.7b-finetuned",
357
  use_auth_token=HF_TOKEN,
358
  )
359
  .half()
 
348
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
349
 
350
  safety_tokenizer = tokenizer = AutoTokenizer.from_pretrained(
351
+ "salexashenko/T5-Base-ROT-epoch-2-train-loss-1.3495-val-loss-1.4164",
352
  use_auth_token=HF_TOKEN,
353
  )
354
  safety_model = model = (
355
  AutoModelForSeq2SeqLM.from_pretrained(
356
+ "salexashenko/T5-Base-ROT-epoch-2-train-loss-1.3495-val-loss-1.4164",
357
  use_auth_token=HF_TOKEN,
358
  )
359
  .half()