sc666 commited on
Commit
dca4367
·
verified ·
1 Parent(s): c71064b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="/data/home/yangxiaoda/FlexCaM/description-video/llamacheckpoint/best_model.pth"
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 其它) 模型生成标签化描述,