Merge pull request #3 from unum-cloud/dev

Fix: Full model names

Author: ashvardanian

README.md

@@ -54,10 +54,12 @@ To load the model:
 ```python
 import uform
 
-model = uform.get_model('english')
-model = uform.get_model('multilingual')
+model = uform.get_model('unum-cloud/uform-vl-english')
+model = uform.get_model('unum-cloud/uform-vl-multilingual')
 ```
 
+You can also load your own Mid-fusion model. Just upload it on HuggingFace and pass model name to `get_model`.
+
 To encode data:
 
 ```python

src/uform.py

@@ -161,7 +161,7 @@ def get_matching_scores(
     def _logit_and_norm(self, embeddings):
         logits = self.clf_head(embeddings)
         if self.head_one_neuron:
-            return torch.sigmoid(logits)
+            return torch.sigmoid(logits)[:, 0]
 
         return F.softmax(logits, dim=1)[:, 1]
 
@@ -336,7 +336,7 @@ def preprocess_text(self, x):
 
 def get_model(model_name, token=None):
     model_path = snapshot_download(
-        repo_id=f'unum-cloud/uform-vl-{model_name}',
+        repo_id=model_name,
         token=token
     )
     config_path = f'{model_path}/config.json'