Future-Tense commited on
Commit
4154b32
1 Parent(s): ad733c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -47,7 +47,8 @@ The function below identifies the device which is availabe to make the predictio
47
  """
48
  def score_frame(frame):
49
  #frame = [torch.tensor(frame)]
50
- results = model(frame)
 
51
  labels = results.xyxyn[0][:, -1].numpy()
52
  cord = results.xyxyn[0][:, :-1].numpy()
53
  return labels, cord
 
47
  """
48
  def score_frame(frame):
49
  #frame = [torch.tensor(frame)]
50
+ result = model(frame)
51
+ results = [torch.tensor(result)]
52
  labels = results.xyxyn[0][:, -1].numpy()
53
  cord = results.xyxyn[0][:, :-1].numpy()
54
  return labels, cord