hantech commited on
Commit
f627d24
β€’
1 Parent(s): b0e9721

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -1,5 +1,4 @@
1
  import pandas as pd
2
- from transformers import pipeline
3
 
4
  import PIL
5
  from PIL import Image
@@ -24,8 +23,15 @@ config['predictor']['beamsearch'] = True
24
  config['device'] = 'cpu' # mps
25
 
26
  recognitor = Predictor(config)
27
- model_name = "microsoft/xdoc-base-squad2.0"
28
- nlp = pipeline('question-answering', model=model_name)
 
 
 
 
 
 
 
29
 
30
  def query(doc, labels):
31
  questions = labels.split(", ")
 
1
  import pandas as pd
 
2
 
3
  import PIL
4
  from PIL import Image
 
23
  config['device'] = 'cpu' # mps
24
 
25
  recognitor = Predictor(config)
26
+ #model_name = "microsoft/xdoc-base-squad2.0"
27
+ #nlp = pipeline('question-answering', model=model_name)
28
+
29
+ from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
30
+
31
+ model_name = "timpal0l/mdeberta-v3-base-squad2"
32
+ model = AutoModelForQuestionAnswering.from_pretrained(model_name)
33
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
34
+ nlp = pipeline('question-answering', model=model, tokenizer=tokenizer)
35
 
36
  def query(doc, labels):
37
  questions = labels.split(", ")