oxkitsune commited on
Commit
157bcb9
·
1 Parent(s): 5f9ec52

set limit back

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -58,6 +58,8 @@ def run_rerun(path_to_video):
58
  if frame_idx % 3 != 0:
59
  continue
60
 
 
 
61
  # resize to avoid excessive time spent processing
62
  frame = cv2.resize(frame, (640, 480))
63
  frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
@@ -93,7 +95,7 @@ def run_rerun(path_to_video):
93
  os.remove(path_to_video)
94
 
95
 
96
- @spaces.GPU(duration=120)
97
  def estimate_depth(frame):
98
  image = transform(frame)
99
  image = image.to(device)
 
58
  if frame_idx % 3 != 0:
59
  continue
60
 
61
+ print("processing frame", frame_idx)
62
+
63
  # resize to avoid excessive time spent processing
64
  frame = cv2.resize(frame, (640, 480))
65
  frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
 
95
  os.remove(path_to_video)
96
 
97
 
98
+ @spaces.GPU(duration=20)
99
  def estimate_depth(frame):
100
  image = transform(frame)
101
  image = image.to(device)