Jyotiyadav commited on
Commit
37d2fcf
·
verified ·
1 Parent(s): df39f4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ def generate_output(input_text):
80
  # Generate outputs
81
  outputs = model.generate(**inputs, max_new_tokens=2500, use_cache=True)
82
  output = tokenizer.batch_decode(outputs)
83
- cleaned_response = process_output(output)
84
  return output,cleaned_response
85
 
86
 
 
80
  # Generate outputs
81
  outputs = model.generate(**inputs, max_new_tokens=2500, use_cache=True)
82
  output = tokenizer.batch_decode(outputs)
83
+ cleaned_response = process_output(str(output))
84
  return output,cleaned_response
85
 
86