letrunglinh commited on
Commit
8102814
1 Parent(s): 3986274

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,11 +1,11 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
- from transformers import AutoModelForQuestionAnswering, 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)
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