Update app.py

app.py

@@ -17,8 +17,6 @@ class MyModel(nn.Module):
     def __init__(self):
         super().__init__()
         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)