ShaoTengLiu commited on
Commit
e496890
1 Parent(s): 40aa755
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -58,13 +58,13 @@ def show_warning(warning_text: str) -> gr.Blocks:
58
  pipe = InferencePipeline(HF_TOKEN)
59
  trainer = Trainer(HF_TOKEN)
60
 
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)
68
 
69
  gr.Markdown(TITLE)
70
  with gr.Tabs():
 
58
  pipe = InferencePipeline(HF_TOKEN)
59
  trainer = Trainer(HF_TOKEN)
60
 
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)
68
 
69
  gr.Markdown(TITLE)
70
  with gr.Tabs():