mashriram commited on
Commit
ccf1da2
·
1 Parent(s): 56d32bd
Files changed (3) hide show
  1. nbeats_model.pt +3 -0
  2. nbeats_model.pt.ckpt +3 -0
  3. stremlit_app.py +7 -0
nbeats_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dfc42f692a0aac8fedf80b08813cbb934ce0df36914c2accb513bc0b83a745b5
3
+ size 74720
nbeats_model.pt.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d33827c0a9bf71172f39c2fde8c19ab5acbbf3868157d2776a3d6645cde4f10
3
+ size 4117890
stremlit_app.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from darts.models import NBEATSModel
3
+
4
+ nbeats = NBEATSModel.load("./nbeats_model.pt")
5
+ df = nbeats.predict(15)
6
+ st.write("Forecast")
7
+ st.write(df.to_json())