Portx
/

trial-multi-model

Model card Files Files and versions

Portx commited on Mar 2, 2025

Commit

7e2c4ad

·

verified ·

1 Parent(s): 30a48ab

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -11,7 +11,7 @@ multi_model_list = [
     {"model_id": "facebook/bart-large-cnn", "task": "summarization"},
     {"model_id": "dslim/bert-base-NER", "task": "token-classification"},
     {"model_id": "textattack/bert-base-uncased-ag-news", "task": "text-classification"},
-    {"model_id": "distilbert/distilbert-base-uncased","task": "question-answering"}
 ]
 class EndpointHandler():

     {"model_id": "facebook/bart-large-cnn", "task": "summarization"},
     {"model_id": "dslim/bert-base-NER", "task": "token-classification"},
     {"model_id": "textattack/bert-base-uncased-ag-news", "task": "text-classification"},
+    {"model_id": "openai-community/gpt2","task": "text-generation"}
 ]
 class EndpointHandler():