Update model.py

model.py

@@ -1,3 +1,16 @@
+
+import numpy as np
+import streamlit as st
+import tensorflow as tf
+import keras
+import matplotlib.pyplot as plt
+import torch
+from skorch import NeuralNetClassifier
+from torch import nn
+import torch.nn.functional as F
+import torchvision.transforms as transforms
+
+
 class Cnn(nn.Module):
     def __init__(self, dropout=0.5):
         super(Cnn, self).__init__()