Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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"
|
| 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):
|
|
|
|
| 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):
|