zac commited on
Commit
19057e5
·
1 Parent(s): 8833e69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ history = []
13
  def generate_text(input_text, history):
14
  print("history ",history)
15
  print("input ", input_text)
16
- output = llm(input_text, max_tokens=1024, stop=["Q:", "\n"], echo=True)
17
  response = output['choices'][0]['text']
18
  return response
19
 
 
13
  def generate_text(input_text, history):
14
  print("history ",history)
15
  print("input ", input_text)
16
+ output = llm(f"Q: {input_text} \n A:", max_tokens=1024, stop=["Q:", "\n"], echo=True)
17
  response = output['choices'][0]['text']
18
  return response
19