Commit
·
ed8c752
1
Parent(s):
b2f3c47
hotfix
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ en_model.tokenizer.add_special_tokens({'additional_special_tokens': ['[POSITIVE-
|
|
| 25 |
en_model.model.resize_token_embeddings(len(en_model.tokenizer))
|
| 26 |
|
| 27 |
def classify(sentiment):
|
| 28 |
-
parameters = {'n_sentences':
|
| 29 |
'batch_size': 2,
|
| 30 |
'avg_len':30,
|
| 31 |
'max_len':50,
|
|
@@ -33,7 +33,7 @@ def classify(sentiment):
|
|
| 33 |
'generation_method':'parallel',
|
| 34 |
'sample': True,
|
| 35 |
'burnin': 450,
|
| 36 |
-
'max_iter':
|
| 37 |
'top_k': 100,
|
| 38 |
'seed_text': f"[{sentiment}-0] [{sentiment}-1] [{sentiment}-2]",
|
| 39 |
# 'verbose': True
|
|
|
|
| 25 |
en_model.model.resize_token_embeddings(len(en_model.tokenizer))
|
| 26 |
|
| 27 |
def classify(sentiment):
|
| 28 |
+
parameters = {'n_sentences': 1,
|
| 29 |
'batch_size': 2,
|
| 30 |
'avg_len':30,
|
| 31 |
'max_len':50,
|
|
|
|
| 33 |
'generation_method':'parallel',
|
| 34 |
'sample': True,
|
| 35 |
'burnin': 450,
|
| 36 |
+
'max_iter': 100,
|
| 37 |
'top_k': 100,
|
| 38 |
'seed_text': f"[{sentiment}-0] [{sentiment}-1] [{sentiment}-2]",
|
| 39 |
# 'verbose': True
|