svjack commited on
Commit
e6037ba
·
verified ·
1 Parent(s): 06a5d67

Update run_abd.py

Browse files
Files changed (1) hide show
  1. run_abd.py +6 -4
run_abd.py CHANGED
@@ -35,10 +35,12 @@ for png_file in tqdm(png_files, desc="Processing images"):
35
 
36
  # 读取.txt文件内容并修改prompt
37
  with open(txt_path, 'r', encoding='utf-8') as f:
38
- prompt = f.read().replace(
39
- "In this digital anime-style drawing,",
40
- "In this realistic personal drawing,"
41
- )
 
 
42
 
43
  # 调用API
44
  result = client.predict(
 
35
 
36
  # 读取.txt文件内容并修改prompt
37
  with open(txt_path, 'r', encoding='utf-8') as f:
38
+ prompt = f.read()
39
+ ####.replace(
40
+ #### "In this digital anime-style drawing,",
41
+ #### "In this realistic personal drawing,"
42
+ ####)
43
+ prompt = ",".join(["In this realistic personal drawing"] + prompt.split(",")[1:])
44
 
45
  # 调用API
46
  result = client.predict(