Lord-Raven commited on
Commit
ae5d8eb
·
1 Parent(s): d43150d

Trying ONNX models on CPU.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import gradio
4
  import json
5
  import time
6
  from datetime import datetime
7
- from transformers import pipeline
8
  from fastapi import FastAPI
9
  from fastapi.middleware.cors import CORSMiddleware
10
  from optimum.onnxruntime import ORTModelForSequenceClassification
 
4
  import json
5
  import time
6
  from datetime import datetime
7
+ from transformers import AutoTokenizer, pipeline
8
  from fastapi import FastAPI
9
  from fastapi.middleware.cors import CORSMiddleware
10
  from optimum.onnxruntime import ORTModelForSequenceClassification