Spaces:

jason-moore
/

sum-soap-demo

Paused

jason-moore commited on Mar 31

Commit

f44fff8

1 Parent(s): cb96fe6

one more

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ Please generate a structured SOAP (Subjective, Objective, Assessment, Plan) note
     generate_ids = model.generate(
         inputs["input_ids"],
-        attention_mask=inputs.attention_mask,  # Explicitly pass attention mask
         max_length=2048,
         num_beams=5,
         no_repeat_ngram_size=2,

     generate_ids = model.generate(
         inputs["input_ids"],
+        attention_mask=inputs["attention_mask"],  # Explicitly pass attention mask
         max_length=2048,
         num_beams=5,
         no_repeat_ngram_size=2,