Arturo Jiménez de los Galanes Reguillos commited on
Commit
efce22a
1 Parent(s): 4157280

Access input properties with .

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,8 +38,8 @@ def translate(python, progress=gr.Progress()):
38
  add_generation_prompt=True,
39
  return_tensors="pt")
40
  inputs = tokenizer(formatted_prompt, return_tensors="pt", padding=True).to(model.device)
41
- attention_mask = inputs["attention_mask"]
42
- input_ids = inputs['input_ids']
43
 
44
  outputs = model.generate(
45
  input_ids,
 
38
  add_generation_prompt=True,
39
  return_tensors="pt")
40
  inputs = tokenizer(formatted_prompt, return_tensors="pt", padding=True).to(model.device)
41
+ attention_mask = inputs.attention_mask
42
+ input_ids = inputs.input_ids
43
 
44
  outputs = model.generate(
45
  input_ids,