Update app.py

app.py

@@ -23,7 +23,7 @@ class TextPredictor(nn.Module):
         return self.fc2(x)
 
 # ====== LOAD SAVED MODEL ======
-checkpoint = torch.load("text_predictor_model.pth", map_location=device)
+checkpoint = torch.load("text_predictor_model.pth", map_location=device, weights_only=False)
 vocab = checkpoint['vocab']
 word2idx = checkpoint['word2idx']
 idx2word = checkpoint['idx2word']