Fralet commited on
Commit
7711d36
·
verified ·
1 Parent(s): 122f7c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ nltk.download('stopwords')
11
  nltk.download('wordnet')
12
 
13
  # Initialize the zero-shot classification pipeline
14
- classifier = pipeline("zero-shot-classification", model="Fralet/personality")
15
 
16
  # Streamlit interface setup
17
  st.title("Resume-based Personality Prediction by Serikov Ayanbek")
 
11
  nltk.download('wordnet')
12
 
13
  # Initialize the zero-shot classification pipeline
14
+ classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
15
 
16
  # Streamlit interface setup
17
  st.title("Resume-based Personality Prediction by Serikov Ayanbek")