polpoDevs commited on
Commit
8f8fb0c
·
verified ·
1 Parent(s): 0005356

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ print(f"Is CUDA available: {torch.cuda.is_available()}")
10
  print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
11
  # Tesla T4
12
 
 
13
  # choose your model here by setting model_chosen_id equal to 1 or 2
14
  model_chosen_id = 2
15
  model_name_options = {
 
10
  print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
11
  # Tesla T4
12
 
13
+
14
  # choose your model here by setting model_chosen_id equal to 1 or 2
15
  model_chosen_id = 2
16
  model_name_options = {