test

app.py

@@ -118,6 +118,20 @@ def process_text(model_name, task, text):
     
     if model_name == 'gpt3.5': 
         if task == 'POS':
+            from transformers import pipeline
+            
+            # Load the pipeline for part-of-speech tagging using the vicuna7b model
+            pos_pipeline = pipeline("ner", model="finiteautomata/bert2crf4ner-pos", tokenizer="finiteautomata/bert2crf4ner-pos")
+            
+            # Define the input phrase
+            input_phrase = "A cat loves the beautiful dog in the neighbor's house"
+            
+            # Use the pipeline to get the part-of-speech tags
+            pos_tags = pos_pipeline(input_phrase)
+            
+            # Print the output
+            print(pos_tags)
+
             strategy1_format = template_all.format(text)
             strategy2_format = prompt2_pos.format(text)
             strategy3_format = demon_pos