Spaces:

ShynBui
/

Vietnamese_correct_spelling

Sleeping

ShynBui commited on Dec 20, 2024

Commit

0f17896

verified ·

1 Parent(s): 16ab1a8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,8 @@ import torch
 print(torch.cuda.is_available())
-summarizer = pipeline("summarization", model="ShynBui/Bartpho_spelling_correction", device=0)
 # @spaces.GPU
 def generate(prompt):

 print(torch.cuda.is_available())
+device = 'cuda' if torch.cuda.is_available() else cpu
+summarizer = pipeline("summarization", model="ShynBui/Bartpho_spelling_correction", device=device)
 # @spaces.GPU
 def generate(prompt):