Szeyu commited on
Commit
908b282
·
verified ·
1 Parent(s): 3093d3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -2,6 +2,13 @@ import streamlit as st
2
  import requests
3
  from bs4 import BeautifulSoup
4
  from transformers import pipeline
 
 
 
 
 
 
 
5
 
6
  # Initialize sentiment analysis pipeline
7
  sentiment_pipeline = pipeline("sentiment-analysis")
 
2
  import requests
3
  from bs4 import BeautifulSoup
4
  from transformers import pipeline
5
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
6
+
7
+ model_id = "LinkLinkWu/Boss_Stock_News_Analysis"
8
+
9
+ # Load tokenizer & Model
10
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
11
+ model = AutoModelForSequenceClassification.from_pretrained(model_id)
12
 
13
  # Initialize sentiment analysis pipeline
14
  sentiment_pipeline = pipeline("sentiment-analysis")