igor commited on
Commit
58b9362
1 Parent(s): 5728ac8
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ llm = Llama.from_pretrained(
21
 
22
  def response(prompt):
23
  llm_response = llm(
24
- "A portrait of a man.", # Prompt
25
  max_tokens=200, # Generate up to 32 tokens, set to None to generate up to the end of the context window
26
  echo=True # Echo the prompt back in the output
27
  )
 
21
 
22
  def response(prompt):
23
  llm_response = llm(
24
+ prompt, # Prompt
25
  max_tokens=200, # Generate up to 32 tokens, set to None to generate up to the end of the context window
26
  echo=True # Echo the prompt back in the output
27
  )