Spaces:
Runtime error
Runtime error
requirements
Browse files- app.py +5 -5
- 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 = "
|
50 |
dict_text_pkl_path = './pkl/dict_text.pkl'
|
51 |
|
52 |
-
image_path = "
|
53 |
model = download_models()
|
54 |
-
text = Text("
|
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="
|
375 |
-
parser.add_argument("--system_prompt", default="
|
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
|