Update app.py

app.py

@@ -20,9 +20,10 @@ def load_model():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     # assert torch.cuda.is_available()
-    tokenizer = AutoTokenizer.from_pretrained("distilbert/distilbert-base-cased")
+    save_dir = "./distilbert/distilbert-base-cased/checkpoint-738"
+    tokenizer = AutoTokenizer.from_pretrained(save_dir)
     model = AutoModelForSequenceClassification.from_pretrained(
-        "./my_model/checkpoint-513"
+        save_dir
     ).to(device)
 
     return tokenizer, model