Update README.md

README.md

@@ -5,7 +5,11 @@ language:
 ---
 # DeeECG EfficientNet V2 Fine Tuned to detect AFIB at 5y
 
-Model compiled with JIT, 
+ - model_name: `efficientnetv2_afib_5y`
+ - Utilizes the EfficientNetV2 architecture for processing ECG signals.
+ - Pre-trained model is loaded from a specified directory, and the model is used to process ECG signal tensors.
+ - The model takes an input signal tensor and outputs the logits for the binary classification of incident AFIB at 5 years.
+ - Model compiled with JIT, 
 
 To load use
 ```