Hugging Face's logo

Spaces:
adriancowham
/
letstalk
Runtime error

App Files Community
Adrian Cowham commited on
Commit
1ccbd42
•
1 Parent(s): 695092a

using finetuned mythomax embedding model

Browse files
Files changed (1) hide show
  1. src/core/embedding.py +1 -1
src/core/embedding.py CHANGED
@@ -49,7 +49,7 @@ class FolderIndex:
49
 
50
 
51
  def embed_files(files: List[File]) -> FolderIndex:
52
- model_name = "adriancowham/letstalk-embed-gte-small"
53
 
54
  model_kwargs = {'device': 'cpu'}
55
  if torch.cuda.is_available():
 
49
 
50
 
51
  def embed_files(files: List[File]) -> FolderIndex:
52
+ model_name = "adriancowham/letstalk-mythomax-embed-gte-small"
53
 
54
  model_kwargs = {'device': 'cpu'}
55
  if torch.cuda.is_available():