nomic-embeddings

Sleeping

derek-thomas HF staff commited on Aug 14

Commit

8a81ae8

•

1 Parent(s): a9f0b60

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,13 +1,11 @@
 from typing import List
 import gradio as gr
-import spaces
 from sentence_transformers import SentenceTransformer
-model = SentenceTransformer("nomic-ai/nomic-embed-text-v1.5", trust_remote_code=True, device='cuda')
-@spaces.GPU
 def embed(document: str):
     return model.encode(document)

 from typing import List
 import gradio as gr
 from sentence_transformers import SentenceTransformer
+model = SentenceTransformer("nomic-ai/nomic-embed-text-v1.5", trust_remote_code=True, device='cpu')
 def embed(document: str):
     return model.encode(document)