Update app.py

app.py

@@ -1,5 +1,5 @@
 import numpy as np
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import AutoTokenizer, AutoModelForSequenceClassification，EsmForSequenceClassification
 from transformers import set_seed
 import torch
 import torch.nn as nn
@@ -16,7 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_checkpoint1)
 class MyModel(nn.Module):
     def __init__(self):
         super().__init__()
-        self.bert1 = AutoModelForSequenceClassification.from_pretrained(model_checkpoint1, num_labels=512)#3000
+        self.bert1 = EsmForSequenceClassification.from_pretrained(model_checkpoint1, num_labels=512)#3000
         # for param in self.bert1.parameters():
         #     param.requires_grad = False
         self.bn1 = nn.BatchNorm1d(256)