Spaces:
Running
Running
mystic_CBK commited on
Commit ·
e548411
1
Parent(s): 6a03c09
🔧 Fix predict endpoint - Use proper fairseq_signals syntax for model inference
Browse files
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(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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)
|