alidev2002 commited on
Commit
e6dc474
·
verified ·
1 Parent(s): ba4adb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -35,9 +35,11 @@ def generate(text, target_language):
35
  return_tensors="pt"
36
  )
37
 
38
- inputs = {k: v.to(model.device) for k, v in tokenized_chat.items()}
39
 
40
- outputs = model.generate(inputs, max_new_tokens=2048)
 
 
41
 
42
  result = tokenizer.decode(outputs[0], skip_special_tokens=True)
43
  return result
 
35
  return_tensors="pt"
36
  )
37
 
38
+ print(tokenized_chat)
39
 
40
+ outputs = model.generate(tokenized_chat, max_new_tokens=2048)
41
+
42
+ print(outputs)
43
 
44
  result = tokenizer.decode(outputs[0], skip_special_tokens=True)
45
  return result