Hugging Face's logo

Spaces:
LinusEats
/
MiniMax-Remover
Build error

App Files Community
PengWeixuanSZU commited on
Commit
6dd3fdf
·
verified ·
1 Parent(s): 2c6d33e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -246,7 +246,7 @@ def track_video(n_frames,video_state):
246
 
247
  global video_predictor
248
  video_predictor=video_predictor.to("cuda")
249
- inference_state = video_predictor.init_state(images=images/255, device=device)
250
  video_state["inference_state"] = inference_state
251
 
252
  if len(torch.from_numpy(video_state["masks"][0]).shape) == 3:
 
246
 
247
  global video_predictor
248
  video_predictor=video_predictor.to("cuda")
249
+ inference_state = video_predictor.init_state(images=images/255, device="cuda")
250
  video_state["inference_state"] = inference_state
251
 
252
  if len(torch.from_numpy(video_state["masks"][0]).shape) == 3: