Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
2 |
import torch
|
|
|
3 |
import streamlit as st
|
4 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
5 |
|
|
|
1 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
2 |
import torch
|
3 |
+
import pickle
|
4 |
import streamlit as st
|
5 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
6 |
|