krushiJyotishii / app.py
mashriram's picture
fix:app.py
3d01ee2
raw
history blame contribute delete
176 Bytes
import streamlit as st
from darts.models import NBEATSModel
nbeats = NBEATSModel.load("./nbeats_model.pt")
df = nbeats.predict(15)
st.write("Forecast")
st.write(df.to_json())