prabinpanta0 commited on
Commit
12add7b
·
verified ·
1 Parent(s): 158f379

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ output: Potential widening of the achievement gap if data is not used equitably.
23
  def generate(text):
24
  try:
25
  prompt = system_prompt + f"\ninput: {text}\noutput:"
26
- inputs = tokenizer.encode(prompt, return_tensors="pt", return_attention_mask=True, pad_token_id=tokenizer.eos_token_id)
27
  outputs = model.generate(inputs["input_ids"], attention_mask=inputs["attention_mask"], max_length=256)
28
  response_text = tokenizer.decode(outputs[0], skip_special_tokens=True).split("output:")[-1].strip()
29
  return response_text if response_text else "No valid response generated."
 
23
  def generate(text):
24
  try:
25
  prompt = system_prompt + f"\ninput: {text}\noutput:"
26
+ inputs = tokenizer.encode(prompt, return_tensors="pt", return_attention_mask=True)
27
  outputs = model.generate(inputs["input_ids"], attention_mask=inputs["attention_mask"], max_length=256)
28
  response_text = tokenizer.decode(outputs[0], skip_special_tokens=True).split("output:")[-1].strip()
29
  return response_text if response_text else "No valid response generated."