sandz7 commited on
Commit
678f5ce
Β·
1 Parent(s): 9633993

reworded the header

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,8 +35,8 @@ def llama_generation(input_text: str,
35
  """
36
 
37
  # Header prompt
38
- header = '''Your are a helpful AI called amphisbeana.
39
- You will help the user, by giving accurate but creative response.\n'''
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')