research14 commited on
Commit
b6b1f92
·
1 Parent(s): 1b46f72
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -94,7 +94,7 @@ with open('demonstration_3_42_parse.txt', 'r') as f:
94
  theme = gr.themes.Soft()
95
 
96
 
97
- gpt_pipeline = pipeline(task="text-generation", model="gpt2")
98
  #vicuna7b_pipeline = pipeline(task="text2text-generation", model="lmsys/vicuna-7b-v1.3")
99
  #vicuna13b_pipeline = pipeline(task="text2text-generation", model="lmsys/vicuna-13b-v1.3")
100
  #vicuna33b_pipeline = pipeline(task="text2text-generation", model="lmsys/vicuna-33b-v1.3")
@@ -109,7 +109,6 @@ model_options = list(model_mapping.keys())
109
  task_options = ['POS', 'Chunking', 'Parsing']
110
 
111
 
112
-
113
  # Function to process text based on model and task
114
  def process_text(model_name, task, text):
115
  gid_list = selected_idx[0:20]
 
94
  theme = gr.themes.Soft()
95
 
96
 
97
+ gpt_pipeline = pipeline(task="text2text-generation", model="gpt2")
98
  #vicuna7b_pipeline = pipeline(task="text2text-generation", model="lmsys/vicuna-7b-v1.3")
99
  #vicuna13b_pipeline = pipeline(task="text2text-generation", model="lmsys/vicuna-13b-v1.3")
100
  #vicuna33b_pipeline = pipeline(task="text2text-generation", model="lmsys/vicuna-33b-v1.3")
 
109
  task_options = ['POS', 'Chunking', 'Parsing']
110
 
111
 
 
112
  # Function to process text based on model and task
113
  def process_text(model_name, task, text):
114
  gid_list = selected_idx[0:20]