fatmata commited on
Commit
9810306
·
verified ·
1 Parent(s): e394d21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -29,16 +29,16 @@ def load_models():
29
  from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequenceClassification
30
 
31
  if tokenizer is None or model is None:
32
- tokenizer = AutoTokenizer.from_pretrained("fatmata/gpt-psybot", use_auth_token=True)
33
- model = AutoModelForCausalLM.from_pretrained("fatmata/gpt-psybot", use_auth_token=True)
34
 
35
  if bert_tokenizer is None or bert_model is None:
36
- bert_tokenizer = AutoTokenizer.from_pretrained("fatmata/bert_model", use_auth_token=True)
37
- bert_model = AutoModelForSequenceClassification.from_pretrained("fatmata/bert_model", use_auth_token=True)
38
 
39
  if tokenizer_c is None or model_c is None:
40
- tokenizer_c = AutoTokenizer.from_pretrained("fatmata/mini_bert", use_auth_token=True)
41
- model_c = AutoModelForSequenceClassification.from_pretrained("fatmata/mini_bert", use_auth_token=True)
42
 
43
  # ---------------- Nettoyage des réponses ----------------
44
  def clean_response(text):
 
29
  from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequenceClassification
30
 
31
  if tokenizer is None or model is None:
32
+ tokenizer = AutoTokenizer.from_pretrained("fatmata/gpt-psybot")
33
+ model = AutoModelForCausalLM.from_pretrained("fatmata/gpt-psybot")
34
 
35
  if bert_tokenizer is None or bert_model is None:
36
+ bert_tokenizer = AutoTokenizer.from_pretrained("fatmata/bert_model")
37
+ bert_model = AutoModelForSequenceClassification.from_pretrained("fatmata/bert_model")
38
 
39
  if tokenizer_c is None or model_c is None:
40
+ tokenizer_c = AutoTokenizer.from_pretrained("fatmata/mini_bert")
41
+ model_c = AutoModelForSequenceClassification.from_pretrained("fatmata/mini_bert")
42
 
43
  # ---------------- Nettoyage des réponses ----------------
44
  def clean_response(text):