chuanli-lambda commited on
Commit
48f2d50
1 Parent(s): 636cfc6

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +2 -5
demo.py CHANGED
@@ -98,12 +98,9 @@ example_videos = gr.components.Dataset(
98
  )
99
 
100
 
101
- args = parse_args()
102
- args.device = "cuda" if torch.cuda.is_available() else "cpu"
103
- print("*** Now using %s." % (args.device))
104
- model = Model(device=args.device)
105
 
106
- with gr.Blocks(theme=args.theme, css="style.css") as demo:
107
  gr.Markdown(DESCRIPTION)
108
 
109
  with gr.Box():
 
98
  )
99
 
100
 
101
+ model = Model(device="cuda")
 
 
 
102
 
103
+ with gr.Blocks(css="style.css") as demo:
104
  gr.Markdown(DESCRIPTION)
105
 
106
  with gr.Box():