Update app.py

app.py

@@ -1,12 +1,11 @@
 import torch
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import FinBERTTokenizer, FinBERTForSequenceClassification
 import yfinance as yf
 import gradio as gr
 
-# Replace "your_model_name" with the actual name of the pre-trained model you want to use
-model_name = "your_model_name"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSequenceClassification.from_pretrained(model_name)
+# Load the FinBERT tokenizer and model
+tokenizer = FinBERTTokenizer.from_pretrained('yiyanghkust/finbert-tone', model_name_or_path='yiyanghkust/finbert-tone')
+model = FinBERTForSequenceClassification.from_pretrained('yiyanghkust/finbert-tone', model_name_or_path='yiyanghkust/finbert-tone')
 
 def predict_stock_price(day, month, year, input_text):
     # Get the stock symbol from the input text