Update app.py

app.py

@@ -2,8 +2,14 @@ import streamlit as st
 import pandas as pd
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import csv
+import random as r
 
-
+with open('train.csv','r') as f:
+    read = csv.reader(f)
+    data = [row for row in read]
+df = pd.DataFrame(data[1:],columns=data[0])
+tweet = df['comment_text'][r.randint(0,1000)]
 tokenizer = AutoTokenizer.from_pretrained("APJ23/MultiHeaded_Sentiment_Analysis_Model")
 model = AutoModelForSequenceClassification.from_pretrained("APJ23/MultiHeaded_Sentiment_Analysis_Model")