jonathanjordan21 commited on
Commit
cf5e8f1
·
verified ·
1 Parent(s): 093b3a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 {"message": f"Model {chosen_model_id} has been succesfully loaded!"}
202
- return {"messsage": f"Model {chosen_model_id} is ready!"}
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,