test

app.py

@@ -27,7 +27,7 @@ syntags = ['NP', 'S', 'VP', 'ADJP', 'ADVP', 'SBAR', 'TOP', 'PP', 'POS', 'NAC', "
  'WHADVP', 'NX', 'PRT', 'VBZ', 'VBP', 'MD', 'NN', 'WHPP', 'SQ', 'SBARQ', 'LST', 'INTJ', 'X', 'UCP', 'CONJP', 'NNP', 'CD', 'JJ',
  'VBD', 'WHADJP', 'PRP', 'RRC', 'NNS', 'SYM', 'CC']
 
-openai.api_key = "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"
+openai.api_key = " "
 
 # determinant vs. determiner
 # https://wikidiff.com/determiner/determinant
@@ -94,8 +94,8 @@ with open('demonstration_3_42_parse.txt', 'r') as f:
 theme = gr.themes.Soft()
 
 
-#gpt_pipeline = pipeline(task="text-generation", model="gpt2")
-vicuna7b_pipeline = pipeline(task="text2text-generation", model="lmsys/vicuna-7b-v1.3")
+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")
 #fastchatT5_pipeline = pipeline(task="text2text-generation", model="lmsys/fastchat-t5-3b-v1.0")
@@ -123,27 +123,27 @@ def process_text(model_name, task, text):
             strategy2 = prompt2_pos.format(text)
             strategy3 = demon_pos
 
-            response1 = vicuna7b_pipeline(strategy1)[0]['generated_text']
-            response2 = vicuna7b_pipeline(strategy2)[0]['generated_text']
-            response3 = vicuna7b_pipeline(strategy3)[0]['generated_text']
+            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 = vicuna7b_pipeline(strategy1)[0]['generated_text']
-            response2 = vicuna7b_pipeline(strategy2)[0]['generated_text']
-            response3 = vicuna7b_pipeline(strategy3)[0]['generated_text']
+            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 = vicuna7b_pipeline(strategy1)[0]['generated_text']
-            response2 = vicuna7b_pipeline(strategy2)[0]['generated_text']
-            response3 = vicuna7b_pipeline(strategy3)[0]['generated_text']
+            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)
 
 # Gradio interface