reworded the header
Browse files
app.py
CHANGED
@@ -35,8 +35,8 @@ def llama_generation(input_text: str,
|
|
35 |
"""
|
36 |
|
37 |
# Header prompt
|
38 |
-
header = '''
|
39 |
-
|
40 |
|
41 |
input_ids = llama_tokenizer.encode(header + input_text,
|
42 |
return_tensors='pt').to('cuda')
|
|
|
35 |
"""
|
36 |
|
37 |
# Header prompt
|
38 |
+
header = '''A conversation between a curious human and an AI assistant called Amphisbeana.
|
39 |
+
The assistant helps the user by giving accurate but creative response.\n\n'''
|
40 |
|
41 |
input_ids = llama_tokenizer.encode(header + input_text,
|
42 |
return_tensors='pt').to('cuda')
|