Update app.py

app.py

@@ -44,4 +44,11 @@ with torch.no_grad():
     outputs = model(**batch)
     predictions = TF.softmax(outputs.logits, dim=1)
     labels = torch.argmax(predictions, dim=1)
-    labels = [model.config.id2label[label_id] for label_id in labels.tolist()]
+    labels = [model.config.id2label[label_id] for label_id in labels.tolist()]
+
+save_directory = "saved"
+tokenizer.save_pretrained(save_directory)
+model.save_pretrained(save_directory)
+
+tokenizer = AutoTokenizer.from_pretrained(save_directory)
+model = AutoModelForSequenceClassification.from_pretrained(save_directory)