Spaces:
Paused
Paused
Commit
·
cb96fe6
1
Parent(s):
749b99f
Dict syntax
Browse files
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
|
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,
|