BlairLeng commited on
Commit
b118f26
1 Parent(s): 2e33d6a

requirements

Browse files
Files changed (2) hide show
  1. app.py +5 -5
  2. requirements.txt +5 -0
app.py CHANGED
@@ -46,12 +46,12 @@ new_directory = os.path.join(current_directory, folder_name)
46
 
47
  pkl_path = './pkl/texts.pkl'
48
  text_image_pkl_path='./pkl/text_image.pkl'
49
- dict_path = "../characters/haruhi/text_image_dict.txt"
50
  dict_text_pkl_path = './pkl/dict_text.pkl'
51
 
52
- image_path = "../characters/haruhi/images"
53
  model = download_models()
54
- text = Text("../characters/haruhi/texts", text_image_pkl_path=text_image_pkl_path,
55
  dict_text_pkl_path=dict_text_pkl_path, model=model, num_steps=50, pkl_path=pkl_path,
56
  dict_path=dict_path, image_path=image_path)
57
 
@@ -371,8 +371,8 @@ class Run:
371
 
372
  if __name__ == '__main__':
373
  parser = argparse.ArgumentParser(description="-----[Chat凉宫春日]-----")
374
- parser.add_argument("--folder", default="../characters/haruhi/texts", help="text folder")
375
- parser.add_argument("--system_prompt", default="../characters/haruhi/system_prompt.txt", help="store system_prompt")
376
  parser.add_argument("--max_len_story", default=1500, type=int)
377
  parser.add_argument("--max_len_history", default=1200, type=int)
378
  # parser.add_argument("--save_path", default="/content/drive/MyDrive/GPTData/Haruhi-Lulu/")
 
46
 
47
  pkl_path = './pkl/texts.pkl'
48
  text_image_pkl_path='./pkl/text_image.pkl'
49
+ dict_path = "characters/haruhi/text_image_dict.txt"
50
  dict_text_pkl_path = './pkl/dict_text.pkl'
51
 
52
+ image_path = "characters/haruhi/images"
53
  model = download_models()
54
+ text = Text("characters/haruhi/texts", text_image_pkl_path=text_image_pkl_path,
55
  dict_text_pkl_path=dict_text_pkl_path, model=model, num_steps=50, pkl_path=pkl_path,
56
  dict_path=dict_path, image_path=image_path)
57
 
 
371
 
372
  if __name__ == '__main__':
373
  parser = argparse.ArgumentParser(description="-----[Chat凉宫春日]-----")
374
+ parser.add_argument("--folder", default="characters/haruhi/texts", help="text folder")
375
+ parser.add_argument("--system_prompt", default="characters/haruhi/system_prompt.txt", help="store system_prompt")
376
  parser.add_argument("--max_len_story", default=1500, type=int)
377
  parser.add_argument("--max_len_history", default=1200, type=int)
378
  # parser.add_argument("--save_path", default="/content/drive/MyDrive/GPTData/Haruhi-Lulu/")
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ transformers
2
+ openai
3
+ tiktoken
4
+ langchain
5
+ gradio