mystic_CBK commited on
Commit
e548411
Β·
1 Parent(s): 6a03c09

πŸ”§ Fix predict endpoint - Use proper fairseq_signals syntax for model inference

Browse files
Files changed (1) hide show
  1. server.py +6 -1
server.py CHANGED
@@ -268,7 +268,12 @@ async def predict_ecg(payload: ECGPayload):
268
  with torch.no_grad():
269
  if fairseq_available:
270
  print("πŸš€ Using fairseq_signals for ECG-FM inference")
271
- result = pretrained_model(signal)
 
 
 
 
 
272
  else:
273
  print("⚠️ Using fallback PyTorch inference")
274
  result = pretrained_model(signal)
 
268
  with torch.no_grad():
269
  if fairseq_available:
270
  print("πŸš€ Using fairseq_signals for ECG-FM inference")
271
+ result = pretrained_model(
272
+ source=signal,
273
+ padding_mask=None,
274
+ mask=False,
275
+ features_only=True
276
+ )
277
  else:
278
  print("⚠️ Using fallback PyTorch inference")
279
  result = pretrained_model(signal)