Update app.py

app.py

@@ -48,9 +48,11 @@ with open(VOCAB_PATH, "rb") as f:
     vocab = pickle.load(f)
 
 # Load model
+device = torch.device("cpu")
 device = torch.device("cpu")  # Change to "cuda" if using GPU
 model = TransformerEncoder(d_model=256, num_heads=1, num_layers=1, vocab_size=len(vocab), max_seq_len=100).to(device)
 model.load_state_dict(torch.load(MODEL_PATH, map_location=device))
+model.to(device)
 model.eval()  # Set model to evaluation mode
 
 print("✅ Model and vocabulary loaded successfully!")