Update handler.py
Browse files- handler.py +1 -2
handler.py
CHANGED
|
@@ -34,9 +34,8 @@ class MultiOutputClassifier(nn.Module):
|
|
| 34 |
return logits
|
| 35 |
|
| 36 |
|
| 37 |
-
class
|
| 38 |
def __init__(self, path: str = ""):
|
| 39 |
-
# ... rest of your code stays the same
|
| 40 |
"""
|
| 41 |
Initialize the handler with the fine-tuned model
|
| 42 |
|
|
|
|
| 34 |
return logits
|
| 35 |
|
| 36 |
|
| 37 |
+
class EndpointHandler(): # Back to original name
|
| 38 |
def __init__(self, path: str = ""):
|
|
|
|
| 39 |
"""
|
| 40 |
Initialize the handler with the fine-tuned model
|
| 41 |
|