Update Inference.py

Inference.py

@@ -46,13 +46,13 @@ def download_file(url, save_path):
             f.write(chunk)
     print(f"✅ {save_path} 저장됨")
 
-DATA_PATH = "converted.jsonl"
+MODEL_PATH = "model.weight.h5"
 TOKENIZER_PATH = "ko_unigram.model"
 
 if not os.path.exists(DATA_PATH):
     download_file(
-        "https://huggingface.co/datasets/Yuchan5386/TinyInst/resolve/main/output.jsonl?download=true",
-        DATA_PATH
+        "https://huggingface.co/Yuchan5386/Model_Prototype/resolve/main/model.weights.h5?download=true",
+        MODEL_PATH
     )
 
 if not os.path.exists(TOKENIZER_PATH):
@@ -189,7 +189,7 @@ model = ReLM(
 dummy_input = np.zeros((1, max_len), dtype=np.int32)
 _ = model(dummy_input)
 model.summary()
-model.load_weights('/kaggle/working/model.weights.h5')
+model.load_weights(MODEL_PATH)
 print("모델 가중치 로드 완료!")
 # =======================
 # 6) 추론 함수 (기존 코드 유지)