Spaces:

m-ric
/

beam_search_visualizer

Running on Zero

m-ric HF staff commited on Mar 21

Commit

46edcf6

•

1 Parent(s): 8f3365d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,8 @@
 import torch
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 # True
-print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 STYLE = """
 .container {

 import torch
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 # True
+if torch.cuda.is_available():
+    print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 STYLE = """
 .container {