JuanJoseMV commited on
Commit
ed8c752
·
1 Parent(s): b2f3c47
Files changed (1) hide show
  1. app.py +2 -2
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': 5,
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': 50,
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