MeYourHint commited on
Commit
9d4e55b
·
1 Parent(s): 2012098
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import os
3
  print("Starting")
4
  import torch
5
  print(f"Is CUDA available: {torch.cuda.is_available()}")
6
- print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
7
  import numpy as np
8
  import gradio as gr
9
  import random
 
3
  print("Starting")
4
  import torch
5
  print(f"Is CUDA available: {torch.cuda.is_available()}")
6
+ # print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
7
  import numpy as np
8
  import gradio as gr
9
  import random