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