Gary3410 commited on
Commit
ca0eb7d
·
1 Parent(s): e7c3fa3

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -43,12 +43,12 @@ def instruct_generate(
43
  temperature: A value controlling the randomness of the sampling process. Higher values result in more random
44
  """
45
  scene_name = os.path.basename(img_path).split(".")[0]
46
- if input in input_value_list:
47
- input = example_dict[scene_name]["input"]
48
  if "..." in input:
49
  input = input.replace("...", "")
50
  output = [prompt, input, max_new_tokens, top_k, temperature]
51
- print(output)
52
  return output
53
 
54
  # 配置具体参数
 
43
  temperature: A value controlling the randomness of the sampling process. Higher values result in more random
44
  """
45
  scene_name = os.path.basename(img_path).split(".")[0]
46
+ # if input in input_value_list:
47
+ # input = example_dict[scene_name]["input"]
48
  if "..." in input:
49
  input = input.replace("...", "")
50
  output = [prompt, input, max_new_tokens, top_k, temperature]
51
+ print(output + scene_name)
52
  return output
53
 
54
  # 配置具体参数