edited responses

app.py

@@ -113,7 +113,7 @@ task_options = ['POS', 'Chunking', 'Parsing']
 
 # Function to process text based on model and task
 def process_text(model_name, task, text):
-    gid_list = selected_idx[0:10]
+    gid_list = selected_idx[0:20]
 
     for gid in tqdm(gid_list, desc='Query'):
             text = ptb[gid]['text']
@@ -123,7 +123,12 @@ def process_text(model_name, task, text):
         strategy2 = task_prompts[task]['strategy2'].format(text)
         strategy3 = task_prompts[task]['strategy3'].format(text)
 
-        return tuple(pipelines[model_name](strategy) for strategy in [strategy1, strategy2, strategy3])
+        # Extract generated text from the model's response
+        response1 = pipelines[model_name](strategy1)[0]['generated_text']
+        response2 = pipelines[model_name](strategy2)[0]['generated_text']
+        response3 = pipelines[model_name](strategy3)[0]['generated_text']
+
+        return response1, response2, response3
 
 # Gradio interface
 iface = gr.Interface(