File size: 318 Bytes
901e561
 
1045950
 
faa6537
5f037c5
933844a
5086aac
5f037c5
535742d
5086aac
4337282
5086aac
4337282
5086aac
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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)