removed prompt

app.py

@@ -94,7 +94,7 @@ with open('demonstration_3_42_parse.txt', 'r') as f:
 theme = gr.themes.Soft()
 
 
-gpt_pipeline = pipeline(task="text2text-generation", model="gpt2")
+gpt_pipeline = pipeline(task="text-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")
@@ -123,27 +123,27 @@ def process_text(model_name, task, text):
             strategy2 = prompt2_pos.format(text)
             strategy3 = demon_pos
 
-            response1 = gpt_pipeline(strategy1)
-            response2 = gpt_pipeline(strategy2)
-            response3 = gpt_pipeline(strategy3)
+            response1 = gpt_pipeline(strategy1)[0]['generated_text']
+            response2 = gpt_pipeline(strategy2)[0]['generated_text']
+            response3 = gpt_pipeline(strategy3)[0]['generated_text']
             return (response1, response2, response3)
         elif task == 'Chunking':
             strategy1 = template_all.format(text)
             strategy2 = prompt2_chunk.format(text)
             strategy3 = demon_chunk
 
-            response1 = gpt_pipeline(strategy1)
-            response2 = gpt_pipeline(strategy2)
-            response3 = gpt_pipeline(strategy3)
+            response1 = gpt_pipeline(strategy1)[0]['generated_text']
+            response2 = gpt_pipeline(strategy2)[0]['generated_text']
+            response3 = gpt_pipeline(strategy3)[0]['generated_text']
             return (response1, response2, response3)
         elif task == 'Parsing':
             strategy1 = template_all.format(text)
             strategy2 = prompt2_parse.format(text)
             strategy3 = demon_parse
             
-            response1 = gpt_pipeline(strategy1)
-            response2 = gpt_pipeline(strategy2)
-            response3 = gpt_pipeline(strategy3)
+            response1 = gpt_pipeline(strategy1)[0]['generated_text']
+            response2 = gpt_pipeline(strategy2)[0]['generated_text']
+            response3 = gpt_pipeline(strategy3)[0]['generated_text']
             return (response1, response2, response3)
     
     # Define prompts for each strategy based on the task