Update handler.py
Browse files- handler.py +1 -0
handler.py
CHANGED
|
@@ -67,3 +67,4 @@ class EndpointHandler:
|
|
| 67 |
probs = torch.nn.functional.softmax(logits.squeeze(0), dim=0)
|
| 68 |
|
| 69 |
return {self.labels[i]: float(probs[i]) for i in range(len(self.labels))}
|
|
|
|
|
|
| 67 |
probs = torch.nn.functional.softmax(logits.squeeze(0), dim=0)
|
| 68 |
|
| 69 |
return {self.labels[i]: float(probs[i]) for i in range(len(self.labels))}
|
| 70 |
+
fix: correct handler indentation
|