Update clip.py

clip.py

@@ -11,7 +11,9 @@ def clip_result(image_path):
     script_dir = os.path.dirname(os.path.abspath(__file__))
     
     # Construct the full path to the file in the subfolder
-    model_path = os.path.join(script_dir, 'model', 'best_clip_model.pth')
+    model_path = os.path.join(script_dir, 'artifacts/models', 'best_clip_model.pth')
+
+    print("model_path:", model_path)
 
     # 載入訓練好的模型和處理器
     model = ChineseCLIPModel.from_pretrained("OFA-Sys/chinese-clip-vit-base-patch16")
@@ -26,7 +28,7 @@ def clip_result(image_path):
     image = Image.open(image_path)
 
     # 2. 加載中文詞彙表
-    with open('/content/drive/MyDrive/chiikawa/word_list.txt', 'r', encoding='utf-8') as f:
+    with open('./chiikawa/word_list.txt', 'r', encoding='utf-8') as f:
         vocab = [line.strip() for line in f.readlines()]
 
     # 3. 圖像和文本處理
@@ -63,8 +65,8 @@ def clip_result(image_path):
 
     # 6. 輸出最接近的前3名中文詞彙
 
-    print("最接近的前3名中文詞彙是:")
-    for rank, word in enumerate(top_k_words, 1):
-        print(f"{rank}. {word}")
+    # print("最接近的前3名中文詞彙是:")
+    # for rank, word in enumerate(top_k_words, 1):
+    #     print(f"{rank}. {word}")
 
     return top_k_words