leygit commited on
Commit
464e4fb
·
verified ·
1 Parent(s): 66b84d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -3,9 +3,13 @@ import pandas as pd
3
  import torch
4
  import torch.nn as nn
5
  import torch.optim as optim
6
- from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
7
  from torch.utils.data import Dataset, DataLoader
 
8
  from sklearn.model_selection import train_test_split
 
 
 
9
 
10
  # Load dataset
11
  file_path = 'spam_ham_dataset.csv'
@@ -103,10 +107,7 @@ with torch.no_grad():
103
  accuracy = correct / total
104
  print(f"Validation Accuracy: {accuracy:.4f}")
105
 
106
- from sklearn.metrics import classification_report
107
- from transformers import BertTokenizer
108
- import torch
109
- import torch.nn.functional as F
110
 
111
  # Classification function
112
  def classify_email(email_text):
 
3
  import torch
4
  import torch.nn as nn
5
  import torch.optim as optim
6
+ import torch.nn.functional as F
7
  from torch.utils.data import Dataset, DataLoader
8
+ from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
9
  from sklearn.model_selection import train_test_split
10
+ from sklearn.metrics import classification_report
11
+ from transformers import BertTokenizer
12
+
13
 
14
  # Load dataset
15
  file_path = 'spam_ham_dataset.csv'
 
107
  accuracy = correct / total
108
  print(f"Validation Accuracy: {accuracy:.4f}")
109
 
110
+
 
 
 
111
 
112
  # Classification function
113
  def classify_email(email_text):