Spaces:
Runtime error
Runtime error
theonerichy
commited on
Commit
•
0aeec0a
1
Parent(s):
f4d5be9
fix requiremetns agane
Browse files- app.py +1 -1
- 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
|