Commit
·
d09388b
1
Parent(s):
bf87a38
test
Browse files
app.py
CHANGED
@@ -10,8 +10,8 @@ args = TTSettings(num_beams=5, min_length=1)
|
|
10 |
tokenizer = T5Tokenizer.from_pretrained("thaboe01/t5-spelling-corrector")
|
11 |
model = T5ForConditionalGeneration.from_pretrained("thaboe01/t5-spelling-corrector")
|
12 |
|
13 |
-
#
|
14 |
-
device = "
|
15 |
model = model.to(device)
|
16 |
|
17 |
# Function to split text into chunks
|
|
|
10 |
tokenizer = T5Tokenizer.from_pretrained("thaboe01/t5-spelling-corrector")
|
11 |
model = T5ForConditionalGeneration.from_pretrained("thaboe01/t5-spelling-corrector")
|
12 |
|
13 |
+
# Ensure the model is on the CPU
|
14 |
+
device = "cpu"
|
15 |
model = model.to(device)
|
16 |
|
17 |
# Function to split text into chunks
|