test

app.py

@@ -87,6 +87,8 @@ if option == 'MILESTONE 3':
     HTCET = [0]*10
     SET = [0]*10
 
+
+
     pred_data = []
 
     for i in range(10):
@@ -100,7 +102,7 @@ if option == 'MILESTONE 3':
                 labels = [model.config.id2label[label_id] for label_id in labels.tolist()]
                 pred_data.append(predictions)
 
-
+    st.write(pred_data[0]['label'])
 
     df = df.assign(Tweet=pre_populated_tweets)
     df = df.assign(Highest_Toxicity_Class_Overall=HTCO)