Update README.md

README.md

@@ -45,10 +45,49 @@ FLAME Large (10M) -- branch FLAME_Large
 You need to install the following packages:
 
 ```shell
-pip install transformers[torch]
+# pip install transformers[torch]
 
-pip install mamba-ssm[causal-conv1d]
+# pip install mamba-ssm[causal-conv1d]
+
+# pip install zuko
+```
+
+To make deterministic or probabilistic forecasts, just follow:
+
+```python
+from transformers import AutoModel, AutoConfig
+import torch
+
+model_path = "path/to/your/model"
+config_path = "path/to/your/config"
+
+config = AutoConfig.from_pretrained(config_path)
+
+model = AutoModel.from_pretrained(model_path, config=config)
+model.eval() 
+
+ # The inputs need to be [batch_size, seq_len]. If multivariate, transform the inputs to [batch_size * n_vars, seq_len]
+inputs = torch.randn(batch_size, seq_length)
+
+# deterministic forecasting
+with torch.no_grad():
+    outputs = model.generate(
+        inputs=inputs,
+        max_length=96,    
+        revin=True,
+        num_samples=1,
+        inference_patch_len=48 # recommend to input the period length
+    )
+
+
+# probabilistic forecasting
+with torch.no_grad():
+    outputs = model.generate(
+        inputs=inputs,
+        max_length=96,    
+        revin=True,
+        num_samples=100
+    )
 
-pip install zuko
 ```