Update app.py

app.py

@@ -9,9 +9,16 @@ from transformers import pipeline, GPT2Tokenizer, AutoTokenizer
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 generator_name_1 = 'MasterAlex69/gpt2_edline_gan'
+generator_name_0 = 'MasterAlex69/gpt2_edline'
+
 generator_tokenizer_1 = GPT2Tokenizer.from_pretrained(generator_name_1)
 generator_tokenizer_1.pad_token_id = generator_tokenizer_1.eos_token_id
+
+generator_tokenizer_0 = GPT2Tokenizer.from_pretrained(generator_name_0)
+generator_tokenizer_0.pad_token_id = generator_tokenizer_0.eos_token_id
+
 generator_pipeline_1 = pipeline('text-generation', model = generator_name_1, tokenizer = generator_tokenizer_1, device = device)
+generator_pipeline_0 = pipeline('text-generation', model = generator_name_0, tokenizer = generator_tokenizer_0, device = device)
 
 ############################################################################################
 
@@ -24,6 +31,10 @@ def generate_text_1():
     result = generator_pipeline_1("Строка состоит из символов", max_length = 225, truncation = False)[0]['generated_text']
     return [result, result]
 
+def generate_text_0():
+    result = generator_pipeline_0("Строка состоит из символов", max_length = 225, truncation = False)[0]['generated_text']
+    return [result, result]
+
 def discriminate_text_1(text):
 
   inputs = discriminator_tokenizer_1(text
@@ -114,7 +125,11 @@ with gr.Blocks(theme = gr.themes.Monochrome()) as iface:
   with gr.Row():
       
     with gr.Column():
-      button_gen_1 = gr.Button("Сгенерировать задание")
+
+      button_gen_0 = gr.Button("Сгенерировать задание (ДО)")
+      button_gen_0_output_text = gr.Textbox(label="Результат генерации", interactive = False)
+        
+      button_gen_1 = gr.Button("Сгенерировать задание (ПОСЛЕ)")
       button_gen_1_output_text = gr.Textbox(label="Результат генерации", interactive = False)
 
     with gr.Column():
@@ -122,6 +137,7 @@ with gr.Blocks(theme = gr.themes.Monochrome()) as iface:
       get_correct_answer_input_text = gr.Textbox(label = "Задание")
       get_correct_answer_output_text = gr.Textbox(label = "Ответ")
 
+  button_gen_0.click(fn = generate_text_0, outputs = [button_gen_0_output_text, get_correct_answer_input_text])
   button_gen_1.click(fn = generate_text_1, outputs = [button_gen_1_output_text, get_correct_answer_input_text])
   button_get_correct_answer.click(fn = get_correct_answer, inputs = get_correct_answer_input_text, outputs = get_correct_answer_output_text)