update

app.py

@@ -20,6 +20,7 @@ import warnings
 from dataclasses import asdict, dataclass
 from typing import Callable, List, Optional
 
+
 import streamlit as st
 import torch
 from torch import nn