Hugging Face's logo

Spaces:
purrgpt-community
/
tiny-purr-1b
Sleeping

App Files Community
FlameF0X commited on
Commit
d623473
·
verified ·
1 Parent(s): e6ddf96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ model = AutoModelForCausalLM.from_pretrained(
14
  )
15
  model.eval()
16
 
17
- purrbert_model = DistilBertForSequenceClassification.from_pretrained("purrgpt-community/PurrBERT-v1")
18
- purrbert_tokenizer = DistilBertTokenizerFast.from_pretrained("purrgpt-community/PurrBERT-v1")
19
  purrbert_model.eval()
20
 
21
  SAFETY_RESPONSE = (
 
14
  )
15
  model.eval()
16
 
17
+ purrbert_model = DistilBertForSequenceClassification.from_pretrained("purrgpt-community/PurrBERT-v1.1")
18
+ purrbert_tokenizer = DistilBertTokenizerFast.from_pretrained("purrgpt-community/PurrBERT-v1.1")
19
  purrbert_model.eval()
20
 
21
  SAFETY_RESPONSE = (