vtiw commited on
Commit
df89d31
1 Parent(s): 5a08d5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -267,7 +267,7 @@ class VectorData():
267
  def __init__(self):
268
  embedding_model_name = 'l3cube-pune/punjabi-sentence-similarity-sbert'
269
 
270
- model_kwargs = {'device':'cuda' if torch.cuda.is_available() else 'cpu',"trust_remote_code": True}
271
 
272
  self.embeddings = HuggingFaceEmbeddings(
273
  model_name=embedding_model_name,
 
267
  def __init__(self):
268
  embedding_model_name = 'l3cube-pune/punjabi-sentence-similarity-sbert'
269
 
270
+ model_kwargs = {'device':'cpu',"trust_remote_code": True}
271
 
272
  self.embeddings = HuggingFaceEmbeddings(
273
  model_name=embedding_model_name,