theonerichy commited on
Commit
0aeec0a
1 Parent(s): f4d5be9

fix requiremetns agane

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def generate_text_instruct(input_text):
9
  response = ""
10
  txt2tag_prompt = f"Provide me with danbooru tags that accurately describe the following description. {input_text}"
11
  for output in llm(f"A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions. USER: {txt2tag_prompt} ASSISTANT: ",
12
- echo=False, stream=True, stop=["</s>", "\n", "User:"]):
13
  answer = output['choices'][0]['text']
14
  response += answer
15
  yield response
 
9
  response = ""
10
  txt2tag_prompt = f"Provide me with danbooru tags that accurately describe the following description. {input_text}"
11
  for output in llm(f"A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions. USER: {txt2tag_prompt} ASSISTANT: ",
12
+ echo=False, stream=True, stop=["</s>", "\n", "User:", "<unk>"]):
13
  answer = output['choices'][0]['text']
14
  response += answer
15
  yield response
requirements.txt CHANGED
@@ -1 +1 @@
1
- llama-cpp-python
 
1
+ llama-cpp-python==0.1.54