Update Inference.py

Inference.py

@@ -46,10 +46,10 @@ def download_file(url, save_path):
             f.write(chunk)
     print(f"✅ {save_path} 저장됨")
 
-MODEL_PATH = "model.weight.h5"
+MODEL_PATH = "model.weights.h5"
 TOKENIZER_PATH = "ko_unigram.model"
 
-if not os.path.exists(DATA_PATH):
+if not os.path.exists(MODEL_PATH):
     download_file(
         "https://huggingface.co/Yuchan5386/Model_Prototype/resolve/main/model.weights.h5?download=true",
         MODEL_PATH
@@ -226,4 +226,4 @@ def generate_text_topp(model, prompt, max_len=512, max_gen=512, p=0.9, temperatu
     return ids_to_text(generated)
 
 print("\n\n===== 생성 결과 =====")  
-print(generate_text_topp(model, "지난 2년 동안", p=0.8))
+print(generate_text_topp(model, "지난 2년 동안", p=0.8))