philipp-zettl
commited on
Commit
•
1444006
1
Parent(s):
88b3b6a
Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
from typing import Dict, List, Any
|
2 |
-
from optimum.
|
3 |
from transformers import AutoTokenizer
|
4 |
from optimum import ORTModelForSeq2SeqLM
|
5 |
|
|
|
1 |
from typing import Dict, List, Any
|
2 |
+
from optimum.pipelines import pipeline
|
3 |
from transformers import AutoTokenizer
|
4 |
from optimum import ORTModelForSeq2SeqLM
|
5 |
|