Spaces:
Sleeping
Sleeping
Singularity666
commited on
Commit
•
982e0a8
1
Parent(s):
9cbba59
Update main.py
Browse files
main.py
CHANGED
@@ -332,4 +332,4 @@ def get_text_embeddings(valid_df):
|
|
332 |
text_embeddings = model.text_projection(text_features)
|
333 |
valid_text_embeddings.append(text_embeddings)
|
334 |
|
335 |
-
return model, torch.cat(valid_text_embeddings)
|
|
|
332 |
text_embeddings = model.text_projection(text_features)
|
333 |
valid_text_embeddings.append(text_embeddings)
|
334 |
|
335 |
+
return model, torch.cat(valid_text_embeddings)
|