final-project / app.py
herztard's picture
model choose
12854bd
raw
history blame
290 Bytes
import streamlit as st
from transformers import pipeline
@st.cache_resource
def load_model():
return pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
model = load_model()
st.title("Tag Detection from CNN News articles")
st.write("Enter a CNN News article URL.")