yasirapunsith commited on
Commit
0f80102
·
1 Parent(s): d49420a

update handler

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -205,7 +205,7 @@ def cluster_windows(predicted_patterns , probability_threshold, window_size,eps
205
  # ========================= locate_patterns function ==========================
206
 
207
  # This will be your primary inference function called by the HF endpoint.
208
- class InferenceHandler:
209
  def __init__(self):
210
  # Model is loaded globally, so it's accessible here
211
  self.model = rocket_model
 
205
  # ========================= locate_patterns function ==========================
206
 
207
  # This will be your primary inference function called by the HF endpoint.
208
+ class EndpointHandler:
209
  def __init__(self):
210
  # Model is loaded globally, so it's accessible here
211
  self.model = rocket_model