Hugging Face's logo

Spaces:
pdltiet
/
demo-gpu
Running on Zero

App Files Community
2
Ayushnangia commited on
Commit
34bc65c
·
verified ·
1 Parent(s): b962691

checking zero gpu presence

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -34,6 +34,8 @@ grammar_args = TTSettings(num_beams=5, min_length=1)
34
  # Spell Check Model initialization
35
  OCRtokenizer = AutoTokenizer.from_pretrained("Bhuvana/t5-base-spellchecker", use_fast=False)
36
  OCRmodel = AutoModelForSeq2SeqLM.from_pretrained("Bhuvana/t5-base-spellchecker")
 
 
37
 
38
 
39
  def correct_spell(inputs):
 
34
  # Spell Check Model initialization
35
  OCRtokenizer = AutoTokenizer.from_pretrained("Bhuvana/t5-base-spellchecker", use_fast=False)
36
  OCRmodel = AutoModelForSeq2SeqLM.from_pretrained("Bhuvana/t5-base-spellchecker")
37
+ zero = torch.Tensor([0]).cuda()
38
+ print(zero.device)
39
 
40
 
41
  def correct_spell(inputs):