fungi commited on
Commit
5b7bd18
1 Parent(s): ac93bcc

use specific model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- qa_pipe = pipeline('question-answering')
 
5
  context = st.text_area('provide context')
6
  question = st.text_area('ask your question')
7
 
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ model_name = 'fungi/deberta-v3-xsmall-squad2'
5
+ qa_pipe = pipeline('question-answering', model=model_name, tokenizer=model_name)
6
  context = st.text_area('provide context')
7
  question = st.text_area('ask your question')
8