Update app.py

app.py

@@ -18,13 +18,14 @@ def load_models():
     # original_model = T5ForConditionalGeneration.from_pretrained(original_model_name).to(device)
 
     # 加载微调模型
-    finetuned_model_path = "utrobinmv/t5_translate_en_ru_zh_small_1024"
-    finetuned_tokenizer = T5Tokenizer.from_pretrained(finetuned_model_path)
-    finetuned_model = T5ForConditionalGeneration.from_pretrained(finetuned_model_path).to(device)
-
-    # finetuned_model_path = r"C:\Users\gdnjr5233_YOLO\Desktop\ВКР_2025\finetuned_model_v2\best_model"
+    # finetuned_model_path = "utrobinmv/t5_translate_en_ru_zh_small_1024"
     # finetuned_tokenizer = T5Tokenizer.from_pretrained(finetuned_model_path)
     # finetuned_model = T5ForConditionalGeneration.from_pretrained(finetuned_model_path).to(device)
+
+    # finetuned_model_path = r"C:\Users\gdnjr5233_YOLO\Desktop\ВКР_2025\finetuned_model_v2\best_model"
+    finetuned_model_path = "finetuned_model_v2/best_model"
+    finetuned_tokenizer = T5Tokenizer.from_pretrained(finetuned_model_path)
+    finetuned_model = T5ForConditionalGeneration.from_pretrained(finetuned_model_path).to(device)
     
     # 加载困惑度模型
     perplexity_model = GPT2LMHeadModel.from_pretrained("gpt2").to(device)