Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -198,8 +198,8 @@ def reload(chosen_model_id):
|
|
| 198 |
model = SentenceTransformer(chosen_model_id, trust_remote_code=True)
|
| 199 |
model_id = chosen_model_id
|
| 200 |
codes_emb = model.encode([x[6:] for x in codes])
|
| 201 |
-
return
|
| 202 |
-
return
|
| 203 |
|
| 204 |
|
| 205 |
|
|
@@ -213,7 +213,7 @@ with gr.Blocks() as demo:
|
|
| 213 |
reload_model = gr.Interface(
|
| 214 |
fn=reload,
|
| 215 |
inputs=[gr.Dropdown(choices=model_ids, value=model_id)],
|
| 216 |
-
outputs="text"
|
| 217 |
# gr.HighlightedText(
|
| 218 |
# label="status",
|
| 219 |
# combine_adjacent=True,
|
|
|
|
| 198 |
model = SentenceTransformer(chosen_model_id, trust_remote_code=True)
|
| 199 |
model_id = chosen_model_id
|
| 200 |
codes_emb = model.encode([x[6:] for x in codes])
|
| 201 |
+
return f"Model {chosen_model_id} has been succesfully loaded!"
|
| 202 |
+
return f"Model {chosen_model_id} is ready!"
|
| 203 |
|
| 204 |
|
| 205 |
|
|
|
|
| 213 |
reload_model = gr.Interface(
|
| 214 |
fn=reload,
|
| 215 |
inputs=[gr.Dropdown(choices=model_ids, value=model_id)],
|
| 216 |
+
outputs="text",
|
| 217 |
# gr.HighlightedText(
|
| 218 |
# label="status",
|
| 219 |
# combine_adjacent=True,
|