amitjohn007 commited on
Commit
7b152ed
·
1 Parent(s): 0d6aa57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -13,9 +13,13 @@ def model_performance_UQS(model_checkpoint, context, question):
13
  if model_checkpoint=="model1":
14
  question_answerer = pipeline("question-answering", model="huggingface/amitjohn007/albert-finetuned-squad")
15
  result=question_answerer(question=question, context=context)
 
16
  elif model_checkpoint=="model2":
17
  question_answerer = pipeline("question-answering", model="huggingface/amitjohn007/bert-finetuned-squad")
18
  result=question_answerer(question=question, context=context)
 
 
 
19
  print("Answer:", result['answer'])
20
  print("Score:", result['score'])
21
  return result
 
13
  if model_checkpoint=="model1":
14
  question_answerer = pipeline("question-answering", model="huggingface/amitjohn007/albert-finetuned-squad")
15
  result=question_answerer(question=question, context=context)
16
+ print("in model1")
17
  elif model_checkpoint=="model2":
18
  question_answerer = pipeline("question-answering", model="huggingface/amitjohn007/bert-finetuned-squad")
19
  result=question_answerer(question=question, context=context)
20
+ print("in model2")
21
+ else:
22
+ print("in default")
23
  print("Answer:", result['answer'])
24
  print("Score:", result['score'])
25
  return result