imperialwool commited on
Commit
580f33f
1 Parent(s): 4a533a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -138,7 +138,7 @@ def chitchatRu():
138
 
139
  inputs = chct_t(text, padding=True, truncation=True, return_tensors="pt")
140
 
141
- generated_ids = chct_m.generate(input_ids=input_ids)
142
  answer = chct_t.decode(generated_ids[0], skip_special_tokens=True)
143
 
144
  return {"status": "pass", "answer": answer}
 
138
 
139
  inputs = chct_t(text, padding=True, truncation=True, return_tensors="pt")
140
 
141
+ generated_ids = chct_m.generate(input_ids=inputs)
142
  answer = chct_t.decode(generated_ids[0], skip_special_tokens=True)
143
 
144
  return {"status": "pass", "answer": answer}