surazbhandari commited on
Commit
88411a9
·
verified ·
1 Parent(s): fe0dabe

Sync from GitHub Actions

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -52,8 +52,8 @@ sys.path.insert(0, model_dir)
52
 
53
  from src.inference import EmbeddingInference
54
 
55
- # Load -- just like sentence-transformers!
56
- model = EmbeddingInference.from_pretrained("surazbhandari/miniembed")
57
 
58
  # 1. Similarity
59
  score = model.similarity("Machine learning is great", "AI is wonderful")
 
52
 
53
  from src.inference import EmbeddingInference
54
 
55
+ # Load model
56
+ model = EmbeddingInference.from_pretrained(model_dir)
57
 
58
  # 1. Similarity
59
  score = model.similarity("Machine learning is great", "AI is wonderful")