ShaoTengLiu commited on
Commit
40aa755
1 Parent(s): 2b4b74e
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ trainer = Trainer(HF_TOKEN)
61
  with gr.Blocks(css='style.css') as demo:
62
  # if SPACE_ID == ORIGINAL_SPACE_ID:
63
  # show_warning(SHARED_UI_WARNING)
64
- elif not torch.cuda.is_available():
65
  show_warning(CUDA_NOT_AVAILABLE_WARNING)
66
  elif (not 'T4' in GPU_DATA):
67
  show_warning(INVALID_GPU_WARNING)
 
61
  with gr.Blocks(css='style.css') as demo:
62
  # if SPACE_ID == ORIGINAL_SPACE_ID:
63
  # show_warning(SHARED_UI_WARNING)
64
+ if not torch.cuda.is_available():
65
  show_warning(CUDA_NOT_AVAILABLE_WARNING)
66
  elif (not 'T4' in GPU_DATA):
67
  show_warning(INVALID_GPU_WARNING)