Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -60,7 +60,7 @@ emotion_prediction_model = AutoModelForSequenceClassification.from_pretrained("b
|
|
60 |
emotion_prediction_tokenizer = AutoTokenizer.from_pretrained("bhadresh-savani/distilbert-base-uncased-emotion")
|
61 |
|
62 |
# Load pre-trained large language model and tokenizer for response generation with increased context window
|
63 |
-
response_model_name = "
|
64 |
response_tokenizer = AutoTokenizer.from_pretrained(response_model_name)
|
65 |
response_model = AutoModelForCausalLM.from_pretrained(response_model_name)
|
66 |
|
|
|
60 |
emotion_prediction_tokenizer = AutoTokenizer.from_pretrained("bhadresh-savani/distilbert-base-uncased-emotion")
|
61 |
|
62 |
# Load pre-trained large language model and tokenizer for response generation with increased context window
|
63 |
+
response_model_name = "gpt2-xl"
|
64 |
response_tokenizer = AutoTokenizer.from_pretrained(response_model_name)
|
65 |
response_model = AutoModelForCausalLM.from_pretrained(response_model_name)
|
66 |
|