commit from

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import numpy as np
 import torch
-from transformers import AutoTokenizer, AutoModel
+from transformers import AutoTokenizer, AutoModel, DistilBertForSequenceClassification
 
 my_model_name = "istassiy/ysda_2022_ml2_hw3_distilbert_base_uncased"
 
@@ -46,7 +46,7 @@ def load_tokenizer():
 
 @st.cache(allow_output_mutation=True)
 def load_model():
-  model = AutoModel.from_pretrained(my_model_name)
+  model = DistilBertForSequenceClassification.from_pretrained(my_model_name)
   return model
 
 def sigmoid(x):