I cannot reproduce the issue using your code snippet and get this output:
model = NeuralNetwork()
x = torch.randn(1, 2)
model.predict(x)
> Predicted data based on trained weights: 
Input (scaled): 
tensor([[1.2890, 0.3778]])
Output: 
tensor([[0.8106]])