Thanks! But shouldnt a simple unsqueeze(0) solve this?
However, i got this error now:
AttributeError: 'list' object has no attribute 'log_softmax'

after i changed
predicted_model[0] to predicted_model

which should basically do the same as unsqueeze.