Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -57,7 +57,7 @@ def parse_tagged_text_to_json(idx, tagged_text):
|
|
| 57 |
def generate_description_json(
|
| 58 |
input_text,
|
| 59 |
llama_model_name="meta-llama/Llama-3.1-8B-Instruct",
|
| 60 |
-
checkpoint_path="
|
| 61 |
):
|
| 62 |
"""
|
| 63 |
读取 input_txt_path 中的描述文本,通过 LLaMA(or 其它) 模型生成标签化描述,
|
|
|
|
| 57 |
def generate_description_json(
|
| 58 |
input_text,
|
| 59 |
llama_model_name="meta-llama/Llama-3.1-8B-Instruct",
|
| 60 |
+
checkpoint_path="best_model.pth"
|
| 61 |
):
|
| 62 |
"""
|
| 63 |
读取 input_txt_path 中的描述文本,通过 LLaMA(or 其它) 模型生成标签化描述,
|