Spaces:
Runtime error
Runtime error
Future-Tense
commited on
Commit
•
4154b32
1
Parent(s):
ad733c4
Update app.py
Browse files
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 |
-
|
|
|
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
|