MojoHz commited on
Commit
550a9e4
·
verified ·
1 Parent(s): db1b4f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -145,7 +145,7 @@ def generate_llama_response(query, context=None):
145
  input_text += f"Context: {context}\n"
146
  input_text += "Answer:"
147
  inputs = tokenizer(input_text, return_tensors="pt")
148
- outputs = model.generate(inputs["input_ids"], max_length=20, temperature=0.7)
149
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
150
  return response
151
 
 
145
  input_text += f"Context: {context}\n"
146
  input_text += "Answer:"
147
  inputs = tokenizer(input_text, return_tensors="pt")
148
+ outputs = model.generate(inputs["input_ids"], max_length=40, temperature=0.7)
149
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
150
  return response
151