Update app.py

app.py

@@ -11,7 +11,7 @@ class RaceClassifier(nn.Module):
 
     def __init__(self, n_classes):
         super(RaceClassifier, self).__init__()
-        self.bert = AutoModel.from_pretrained("vinai/bertweet-large")
+        self.bert = AutoModel.from_pretrained("vinai/bertweet-base")
         self.drop = nn.Dropout(p=0.3)  # can be changed in future
         self.out = nn.Linear(self.bert.config.hidden_size,
                              n_classes)  # linear layer for the output with the number of classes
@@ -104,7 +104,7 @@ def predict(*text):
     print(tweets)
     sentences = tweets
 
-    tokenizer = AutoTokenizer.from_pretrained("vinai/bertweet-large", normalization=True)
+    tokenizer = AutoTokenizer.from_pretrained("vinai/bertweet-base", normalization=True)
 
     encoded_sentences = tokenizer(
         sentences,