Getting validation error caused by HTTPConectionPool Read Timed Out.

I'm currently trying to load my PyTorch model to Label Studio. I've tried loading this with another string matching-based model which worked fine (although I got some Assertion Errors as mentioned in https://github.com/heartexlabs/label-studio-ml-backend/issues/117).

The problem I'm facing is that when I try to add a new model via the Label Studio UI, I get the following:

```
Validation error
Successfully connected to $HOST but it doesn't look like a valid ML backend. Reason: HTTPConnectionPool(host=$HOST, port=$POST): Read timed out. (read timeout=3.0).
Check the ML backend server console logs to check the status. There might be
something wrong with your model or it might be incompatible with the current labeling configuration.
```

I've followed the HuggingFace example mainly and made the appropriate changes to fit my use case. Any idea on what the problem might be?