erikacardenas300's picture
Update app.py
535742d
raw
history blame contribute delete
318 Bytes
import os
os.system('pip install transformers')
os.system('pip install torch')
import torch
import streamlit as st
from transformers import pipeline
pipe = pipeline("text-classification",model="erikacardenas300/Company_Classifier")
text = st.text_area('Enter text')
if text:
out = pipe(text)
st.json(out)