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)
         |