test table

app.py

@@ -89,30 +89,30 @@ if option == 'MILESTONE 3':
 
 
 
-    pred_data = []
+    # pred_data = []
+
+    # for i in range(10):
+    #     X_train = pre_populated_tweets[i]
+    #     for j in range(6):
+    #         batch = tokenizers[j](X_train, padding=True, truncation=True, max_length=512, return_tensors="pt")
+    #         with torch.no_grad():
+    #             outputs = models[j](**batch, labels=torch.tensor([1, 0]))
+    #             predictions = F.softmax(outputs.logits, dim=1)
+    #             labels = torch.argmax(predictions, dim=1)
+    #             labels = [model.config.id2label[label_id] for label_id in labels.tolist()]
+    #             pred_data.append(predictions)
+
+    # st.write(pred_data[0]['label'])
 
     for i in range(10):
-        X_train = pre_populated_tweets[i]
-        for j in range(6):
-            batch = tokenizers[j](X_train, padding=True, truncation=True, max_length=512, return_tensors="pt")
-            with torch.no_grad():
-                outputs = models[j](**batch, labels=torch.tensor([1, 0]))
-                predictions = F.softmax(outputs.logits, dim=1)
-                labels = torch.argmax(predictions, dim=1)
-                labels = [model.config.id2label[label_id] for label_id in labels.tolist()]
-                pred_data.append(predictions)
-
-    st.write(pred_data[0]['label'])
-
-for i in range(10):
-    HTCO[i]=pred_data[i]['label']
-
-    df = df.assign(Tweet=pre_populated_tweets)
-    df = df.assign(Highest_Toxicity_Class_Overall=HTCO)
-    df = df.assign(Score_Overall=SO)
-    df = df.assign(Highest_Toxicity_Class_Except_Toxic=HTCET)
-    df = df.assign(Score_Except_Toxic=SET)
-        
+        HTCO[i]=pred_data[i]['label']
+
+        df = df.assign(Tweet=pre_populated_tweets)
+        df = df.assign(Highest_Toxicity_Class_Overall=HTCO)
+        df = df.assign(Score_Overall=SO)
+        df = df.assign(Highest_Toxicity_Class_Except_Toxic=HTCET)
+        df = df.assign(Score_Except_Toxic=SET)
+            
     st.table(df)
 
     st.write('test2')