test2

app.py

@@ -118,32 +118,37 @@ def process_text(model_name, task, text):
     
     if model_name == 'vicuna-7b': 
         if task == 'POS':
-            strategy1 = template_all.format(text)
-            strategy2 = prompt2_pos.format(text)
-            strategy3 = demon_pos
-
-            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)
+            strategy1_format = template_all.format(text)
+            strategy2_format = prompt2_pos.format(text)
+            strategy3_format = demon_pos
+
+            result1 = gpt_pipeline(strategy1_format)
+            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']
+
+            return (generated_text1, generated_text2, generated_text3)
         elif task == 'Chunking':
-            strategy1 = template_all.format(text)
-            strategy2 = prompt2_chunk.format(text)
-            strategy3 = demon_chunk
-
-            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)
+            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 = template_all.format(text)
-            strategy2 = prompt2_parse.format(text)
-            strategy3 = demon_parse
+            strategy1_format = template_all.format(text)
+            strategy2_format = prompt2_parse.format(text)
+            strategy3_format = demon_parse
             
-            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)
+            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(