Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -32,7 +32,7 @@ class EndpointHandler:
|
|
32 |
|
33 |
inputs = data.get("inputs")
|
34 |
videos = read_video(inputs)
|
35 |
-
outputs = model(videos)
|
36 |
_, predicted = torch.max(outputs.logits, 1)
|
37 |
return predicted
|
38 |
|
|
|
32 |
|
33 |
inputs = data.get("inputs")
|
34 |
videos = read_video(inputs)
|
35 |
+
outputs = self.model(videos)
|
36 |
_, predicted = torch.max(outputs.logits, 1)
|
37 |
return predicted
|
38 |
|