update

app.py

@@ -5,6 +5,7 @@ from transformers import AutoTokenizer, AutoModel, MarianMTModel, MarianTokenize
 from sklearn.metrics.pairwise import cosine_similarity
 import re
 from huggingface_hub import hf_hub_download
+import torch
 # ========== 模型加载 ==========
 embedding_model_name = "BAAI/bge-small-en"
 tokenizer = AutoTokenizer.from_pretrained(embedding_model_name)

requirements.txt

@@ -3,4 +3,4 @@ transformers
 gradio
 scikit-learn
 sentencepiece
-huggingface_hub
+huggingface_hub