Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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])
|