nppmatt commited on
Commit
748a83f
1 Parent(s): b6de9b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import pandas as pd
3
  import torch
4
  from torch import nn
5
  from torch.utils.data import Dataset, DataLoader
6
- from transformers import AutoTokenizer, BertModel
7
  from sklearn import metrics
8
  import streamlit as st
9
 
 
3
  import torch
4
  from torch import nn
5
  from torch.utils.data import Dataset, DataLoader
6
+ from transformers import AutoTokenizer, BertForSequenceClassification
7
  from sklearn import metrics
8
  import streamlit as st
9