adding special tokens

app.py

@@ -21,8 +21,11 @@ tokenizer = en_model.tokenizer
 model = en_model.model
 device = model.device
 
+en_model.tokenizer.add_special_tokens({'additional_special_tokens': ['[POSITIVE-0]', '[POSITIVE-1]', '[POSITIVE-2]','[NEGATIVE-0]', '[NEGATIVE-1]', '[NEGATIVE-2]']})
+en_model.model.resize_token_embeddings(len(en_model.tokenizer))
+
 def classify(sentiment):
-    parameters = {'n_sentences': 10,  
+    parameters = {'n_sentences': 5,  
               'batch_size': 2,
               'avg_len':30,
               'max_len':50,
@@ -30,7 +33,7 @@ def classify(sentiment):
               'generation_method':'parallel',
               'sample': True,
               'burnin': 450,
-              'max_iter': 500,
+              'max_iter': 150,
               'top_k': 100,
               'seed_text': f"[{sentiment}-0] [{sentiment}-1] [{sentiment}-2]",
             #   'verbose': True
@@ -44,7 +47,7 @@ demo = gr.Blocks()
 
 with demo:
     gr.Markdown()
-    inputs = gr.Dropdown(value=["POSITIVE", "NEGATIVE"], label="Sentiment to generate")
+    inputs = gr.Dropdown(["POSITIVE", "NEGATIVE"], label="Sentiment to generate")
     output = gr.Textbox(label="Generated tweet")
     b1 = gr.Button("Generate")
     b1.click(classify, inputs=inputs, outputs=output)