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])
|