jason-moore commited on
Commit
cb96fe6
·
1 Parent(s): 749b99f

Dict syntax

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ Please generate a structured SOAP (Subjective, Objective, Assessment, Plan) note
47
  inputs = {k: v.to(device) for k, v in inputs.items()}
48
 
49
  generate_ids = model.generate(
50
- inputs.input_ids,
51
  attention_mask=inputs.attention_mask, # Explicitly pass attention mask
52
  max_length=2048,
53
  num_beams=5,
 
47
  inputs = {k: v.to(device) for k, v in inputs.items()}
48
 
49
  generate_ids = model.generate(
50
+ inputs["input_ids"],
51
  attention_mask=inputs.attention_mask, # Explicitly pass attention mask
52
  max_length=2048,
53
  num_beams=5,