jonathanjordan21 commited on
Commit
76ee9e7
·
verified ·
1 Parent(s): eb89c17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -397,7 +397,7 @@ def load_model(model_id):
397
  automodel_args={"torch_dtype": "auto"},
398
  trust_remote_code=True,
399
  )
400
- return model, _
401
  else:
402
  model = SentenceTransformer(model_id, trust_remote_code=True)
403
  # codes_emb = model.encode([x[6:] for x in codes])
 
397
  automodel_args={"torch_dtype": "auto"},
398
  trust_remote_code=True,
399
  )
400
+ return model, None
401
  else:
402
  model = SentenceTransformer(model_id, trust_remote_code=True)
403
  # codes_emb = model.encode([x[6:] for x in codes])