Update app.py

app.py

@@ -1,10 +1,18 @@
 import streamlit as st
 from transformers import pipeline
+import torch
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-pipe = pipeline("sentiment-analysis")
+
+tokenizer = AutoTokenizer.from_pretrained("nickmuchi/sec-bert-finetuned-finance-classification")
+model = AutoModelForSequenceClassification.from_pretrained("nickmuchi/sec-bert-finetuned-finance-classification")
+
+
+pipe = pipeline("text-classification", model=model, tokenizer=tokenizer, device=device)
 text = st.text_area("Enter some text")
 
 if text:
     out = pipe(text)
-    st.json(out)
+    st.json(out)