Upload folder using huggingface_hub

app.py

@@ -108,7 +108,7 @@ def main(args):
     model = SentenceTransformer(
         args.model_name_or_path, 
         revision=args.revision,
-        backend="openvino")
+        )
 
     document_state = gr.State()
 
@@ -122,6 +122,7 @@ def main(args):
                    analytics_enabled=False) as demo:
         gr.Markdown(file_example)
         doc_files_box = gr.File(label="Upload Documents", file_types=[".json"], file_count="single")
+        model_selection = gr.Radio(["HIT-TMG/KaLM-embedding-multilingual-mini-instruct-v1.5"], value="HIT-TMG/KaLM-embedding-multilingual-mini-instruct-v1.5", label="Model Selection", interactive=False)
         retrieval_interface = gr.Interface(
             fn=retrieve_document_fn,
             inputs=[gr.Textbox(label="Query"), document_state],
@@ -152,7 +153,6 @@ def main(args):
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument("--model_name_or_path", type=str, default="HIT-TMG/KaLM-embedding-multilingual-mini-instruct-v1.5")
-    # parser.add_argument("--model_name_or_path", type=str, default="/raid/hxs/Checkpoints/huggingface_models/bge-base-en-v1.5")
     parser.add_argument("--revision", type=str, default="refs/pr/2")
 
     args = parser.parse_args()