anonymous5378 commited on
Commit
4b69a05
·
verified ·
1 Parent(s): 4d90972

Fix pipeline model_id

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def predict(title, type, base_model, dataset, num_beams):
16
  elif type == "Pipeline":
17
  ag_model_id = directory + f"{base_model.lower()}-ag-{dataset.lower()}"
18
  ve_model_id = directory + f"{base_model.lower()}-ve-{dataset.lower()}"
19
- model = TransformersAVG(ag_model, model_ve=ve_model)
20
  predictions = model.generate_av_pipeline(title, num_beams=num_beams)
21
  elif type == 'Multitask':
22
  model_id = directory + f"{base_model.lower()}-mlt-{dataset.lower()}"
 
16
  elif type == "Pipeline":
17
  ag_model_id = directory + f"{base_model.lower()}-ag-{dataset.lower()}"
18
  ve_model_id = directory + f"{base_model.lower()}-ve-{dataset.lower()}"
19
+ model = TransformersAVG(ag_model_id, model_ve=ve_model_id)
20
  predictions = model.generate_av_pipeline(title, num_beams=num_beams)
21
  elif type == 'Multitask':
22
  model_id = directory + f"{base_model.lower()}-mlt-{dataset.lower()}"