Update app.py

app.py

@@ -2,11 +2,11 @@ import gradio as gr
 import json
 from sentence_transformers import SentenceTransformer
 
-model = SentenceTransformer('all-MiniLM-L6-v2', device='cuda')
+model = SentenceTransformer('all-MiniLM-L6-v2')
 model.max_seq_length = 256
 
 def text_to_embedding(text):
-    embedding = model.encode(text, device='cuda')
+    embedding = model.encode(text)
     return json.dumps(embedding)
 
 inputs = gr.inputs.Textbox(default="Type text here.")