Update handler.py
Browse files- handler.py +1 -0
handler.py
CHANGED
|
@@ -11,6 +11,7 @@ multi_model_list = [
|
|
| 11 |
{"model_id": "facebook/bart-large-cnn", "task": "summarization"},
|
| 12 |
{"model_id": "dslim/bert-base-NER", "task": "token-classification"},
|
| 13 |
{"model_id": "textattack/bert-base-uncased-ag-news", "task": "text-classification"},
|
|
|
|
| 14 |
]
|
| 15 |
|
| 16 |
class EndpointHandler():
|
|
|
|
| 11 |
{"model_id": "facebook/bart-large-cnn", "task": "summarization"},
|
| 12 |
{"model_id": "dslim/bert-base-NER", "task": "token-classification"},
|
| 13 |
{"model_id": "textattack/bert-base-uncased-ag-news", "task": "text-classification"},
|
| 14 |
+
{"model_id": "distilbert/distilbert-base-uncased","task": "question-answering"}
|
| 15 |
]
|
| 16 |
|
| 17 |
class EndpointHandler():
|