Update app.py

app.py

@@ -16,14 +16,14 @@ tokenizer = AutoTokenizer.from_pretrained(model_checkpoint1)
 class MyModel(nn.Module):
     def __init__(self):
         super().__init__()
-        self.bert1 = EsmForSequenceClassification.from_pretrained(model_checkpoint1, num_labels=512)  # 3000
+        self.bert1 = EsmForSequenceClassification.from_pretrained(model_checkpoint1, num_labels=3000)  # 3000
         # for param in self.bert1.parameters():
         #     param.requires_grad = False
         self.bn1 = nn.BatchNorm1d(256)
         self.bn2 = nn.BatchNorm1d(128)
         self.bn3 = nn.BatchNorm1d(64)
         self.relu = nn.LeakyReLU()
-        self.fc1 = nn.Linear(512, 256)
+        self.fc1 = nn.Linear(3000, 256)
         self.fc2 = nn.Linear(256, 128)
         self.fc3 = nn.Linear(128, 64)
         self.output_layer = nn.Linear(64, 2)