Update app.py

app.py

@@ -99,19 +99,10 @@ def respond(
     message,
     history: list[tuple[str, str]],
     threshold,
-    is_multiple,
-    chosen_model_id
+    is_multiple
 ):
     global codes_emb
     global undetected
-    global model
-    global model_id
-    global codes_emb
-
-    if chosen_model_id != model_id:
-        model = SentenceTransformer(chosen_model_id, trust_remote_code=True)
-        model_id = chosen_model_id
-        codes_emb = model.encode([x[6:] for x in codes])
 
     undetected_code = undetected[:3]
 
@@ -198,6 +189,20 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 # )
 
 
+def reload(chosen_model_id):
+    global model
+    global model_id
+    global codes_emb
+
+    if chosen_model_id != model_id:
+        model = SentenceTransformer(chosen_model_id, trust_remote_code=True)
+        model_id = chosen_model_id
+        codes_emb = model.encode([x[6:] for x in codes])
+        return {"message": f"Model {chosen_model_id} has been succesfully loaded!"}
+    return {"messsage": f"Model {chosen_model_id} is ready!"}
+    
+
+
 with gr.Blocks() as demo:
     # Add header title and description
     gr.Markdown("# List of Request Numbers")
@@ -205,13 +210,15 @@ with gr.Blocks() as demo:
     gr.Markdown("# Valid License Plate Number Criteria:")
     gr.Markdown("(1-2 letters) (4 numbers) (1-3 letters)")
 
+    reload_model = gr.Interface(fn=image_classifier, inputs=[gr.Dropdown(choices=model_ids, value=model_id)], outputs="label")
+
     # Add chat interface
     chat_interface = gr.ChatInterface(
         respond, 
         additional_inputs=[
             gr.Number(0.5, label="confidence threshold", show_label=True, minimum=0., maximum=1.0, step=0.1),
             gr.Checkbox(label="multiple", info="Allow multiple request code numbers"),
-            gr.Dropdown(choices=model_ids, value=model_id)
+            
         ]
     )