Spaces:
Paused
Paused
Commit
·
749b99f
1
Parent(s):
34b3b8f
Missed one
Browse files
app.py
CHANGED
|
@@ -33,7 +33,8 @@ Please generate a structured SOAP (Subjective, Objective, Assessment, Plan) note
|
|
| 33 |
|
| 34 |
{doctor_patient_conversation}
|
| 35 |
<|assistant|>"""
|
| 36 |
-
|
|
|
|
| 37 |
# Tokenize and generate with explicit padding settings
|
| 38 |
inputs = tokenizer(
|
| 39 |
prompt,
|
|
|
|
| 33 |
|
| 34 |
{doctor_patient_conversation}
|
| 35 |
<|assistant|>"""
|
| 36 |
+
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 37 |
+
|
| 38 |
# Tokenize and generate with explicit padding settings
|
| 39 |
inputs = tokenizer(
|
| 40 |
prompt,
|