Spaces:
Build error
Build error
Fix pipeline model_id
Browse files
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(
|
| 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()}"
|