renamed api.py->app.py

classifier/Albert_latest.py

@@ -4,7 +4,7 @@ import torch
 class Model:
     def __init__(self, model_weights):
         self.tokenizer = AlbertTokenizer.from_pretrained('albert-base-v2')
-        self.model = AlbertForSequenceClassification.from_pretrained('albert-base-v2', num_labels=2)
+        self.model = AlbertForSequenceClassification.from_pretrained('albert-base-v2', num_labels=5)
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         
         # Load the checkpoint