Spaces:
Paused
Paused
letrunglinh
commited on
Commit
•
8102814
1
Parent(s):
3986274
Update app.py
Browse files
app.py
CHANGED
@@ -1,11 +1,11 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
-
from transformers import
|
4 |
from optimum.intel import OVModelForQuestionAnswering
|
5 |
model_checkpoint = "letrunglinh/qa_pnc"
|
6 |
global model_convert,tokenizer
|
7 |
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
|
8 |
-
model_convert = OVModelForQuestionAnswering.from_pretrained(model_checkpoint,export=True)
|
9 |
model_convert.half()
|
10 |
model_convert.compile()
|
11 |
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
+
from transformers import pipeline,AutoTokenizer
|
4 |
from optimum.intel import OVModelForQuestionAnswering
|
5 |
model_checkpoint = "letrunglinh/qa_pnc"
|
6 |
global model_convert,tokenizer
|
7 |
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
|
8 |
+
model_convert = OVModelForQuestionAnswering.from_pretrained(model_checkpoint,export=True, compile=False)
|
9 |
model_convert.half()
|
10 |
model_convert.compile()
|
11 |
|