Update app.py

app.py

@@ -10,7 +10,7 @@ import time
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
 
-tok = GPT2Tokenizer()
+tok = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
 model_checkpoint = 'cointegrated/rubert-tiny-toxicity'
 toxicity_tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 toxicity_model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint)
@@ -145,6 +145,8 @@ def main():
         page_reviews_classification()
     elif page_selection == "Анализ токсичности":
         page_toxicity_analysis()
+    elif page_selection == "Генерация текста Noize MC":
+        page_gpt_generation()
 
 if __name__ == "__main__":
     main()