Update app.py

app.py

@@ -15,7 +15,7 @@ tokenizer = get_tokenizer()
 class devops_model(nn.Module):
     def __init__(self):
         super(devops_model, self).__init__()
-        self.bert = None
+        self.berta = None
         self.fc = nn.Sequential(
             nn.Linear(768, 768),
             nn.ReLU(),
@@ -26,7 +26,7 @@ class devops_model(nn.Module):
         )
         
     def forward(self, train_batch):
-        emb = self.bert(**train_batch)['last_hidden_state'].mean(axis=1)
+        emb = self.berta(**train_batch)['last_hidden_state'].mean(axis=1)
         return self.fc(emb)
 
 @st.cache