test3

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("fill-mask", 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")
@@ -131,24 +131,24 @@ def process_text(model_name, task, text):
             generated_text3 = result3[0]['sequence']
 
             return (generated_text1, generated_text2, generated_text3)
-        elif task == 'Chunking':
-            strategy1_format = template_all.format(text)
-            strategy2_format = prompt2_chunk.format(text)
-            strategy3_format = demon_chunk
-
-            result1 = gpt_pipeline(strategy1_format)[0]['generated_text']
-            result2 = gpt_pipeline(strategy2_format)[0]['generated_text']
-            result3 = gpt_pipeline(strategy3_format)[0]['generated_text']
-            return (result1, result2, result3)
-        elif task == 'Parsing':
-            strategy1_format = template_all.format(text)
-            strategy2_format = prompt2_parse.format(text)
-            strategy3_format = demon_parse
+        # elif task == 'Chunking':
+        #     strategy1_format = template_all.format(text)
+        #     strategy2_format = prompt2_chunk.format(text)
+        #     strategy3_format = demon_chunk
+
+        #     result1 = gpt_pipeline(strategy1_format)[0]['generated_text']
+        #     result2 = gpt_pipeline(strategy2_format)[0]['generated_text']
+        #     result3 = gpt_pipeline(strategy3_format)[0]['generated_text']
+        #     return (result1, result2, result3)
+        # elif task == 'Parsing':
+        #     strategy1_format = template_all.format(text)
+        #     strategy2_format = prompt2_parse.format(text)
+        #     strategy3_format = demon_parse
             
-            result1 = gpt_pipeline(strategy1_format)[0]['generated_text']
-            result2 = gpt_pipeline(strategy2_format)[0]['generated_text']
-            result3 = gpt_pipeline(strategy3_format)[0]['generated_text']
-            return (result1, result2, result3)
+        #     result1 = gpt_pipeline(strategy1_format)[0]['generated_text']
+        #     result2 = gpt_pipeline(strategy2_format)[0]['generated_text']
+        #     result3 = gpt_pipeline(strategy3_format)[0]['generated_text']
+        #     return (result1, result2, result3)
 
 # Gradio interface
 iface = gr.Interface(