zszazi

import streamlit as st
import requests
import json 
from PIL import Image
import pandas as pd
import numpy as np
import wget
import _pickle as cPickle

st.title('Predictive Maintenance - TTF (total time to failure)')

zszazi

class NotificationCallback(Callback):
    """
        PyTorch callback for model training
    """
    def on_train_begin(self, metrics_names: StrList, **kwargs: Any) -> None:
        send_notification("Training has Begun ")

        
    def on_epoch_end(self,  epoch: int, smooth_loss: Tensor, last_metrics: MetricsList, **kwargs: Any) -> None:
        super().on_epoch_end(**kwargs)