herztard commited on
Commit
12854bd
·
1 Parent(s): b2204f0

model choose

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -1,4 +1,11 @@
1
  import streamlit as st
 
 
 
 
2
 
3
- x = st.slider('Select a value')
4
- st.write(x, 'squared is', x * x)
 
 
 
 
1
  import streamlit as st
2
+ from transformers import pipeline
3
+ @st.cache_resource
4
+ def load_model():
5
+ return pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
6
 
7
+ model = load_model()
8
+
9
+
10
+ st.title("Tag Detection from CNN News articles")
11
+ st.write("Enter a CNN News article URL.")