armanokka commited on
Commit
f47b4ba
1 Parent(s): b3c6542

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,9 +5,9 @@ import time
5
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
6
  from flores200_codes import flores_codes
7
 
8
- print(f"Is CUDA available: {torch.cuda.is_available()}")
9
  # True
10
- print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
11
 
12
  def load_models():
13
  # build model and tokenizer
 
5
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
6
  from flores200_codes import flores_codes
7
 
8
+ #print(f"Is CUDA available: {torch.cuda.is_available()}")
9
  # True
10
+ #print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
11
 
12
  def load_models():
13
  # build model and tokenizer