test

app.py

@@ -94,7 +94,7 @@ with open('demonstration_3_42_parse.txt', 'r') as f:
 theme = gr.themes.Soft()
 
 
-gpt_pipeline = pipeline("fill-mask", model="gpt2")
+gpt_pipeline = pipeline(task="text2text-generation", model="gpt2")
 #vicuna7b_pipeline = pipeline(task="text2text-generation", model="lmsys/vicuna-7b-v1.3")
 #vicuna13b_pipeline = pipeline(task="text2text-generation", model="lmsys/vicuna-13b-v1.3")
 #vicuna33b_pipeline = pipeline(task="text2text-generation", model="lmsys/vicuna-33b-v1.3")
@@ -126,9 +126,9 @@ def process_text(model_name, task, text):
             result2 = gpt_pipeline(strategy2_format)
             result3 = gpt_pipeline(strategy3_format)
 
-            generated_text1 = result1[0]['sequence']
-            generated_text2 = result2[0]['sequence']
-            generated_text3 = result3[0]['sequence']
+            generated_text1 = result1[0]['generated_text']
+            generated_text2 = result2[0]['generated_text']
+            generated_text3 = result3[0]['generated_text']
 
             return (generated_text1, generated_text2, generated_text3)
         # elif task == 'Chunking':