Hugging Face's logo

Spaces:
nullHawk
/
digit_recognition_ffnn
Sleeping

App Files Community
nullHawk commited on
Commit
a6ef0e5
·
verified ·
1 Parent(s): 78f38b4

added requirments.txt

Browse files
Files changed (4) hide show
  1. .gitignore +1 -2
  2. model/model.pt +3 -0
  3. requirments.txt +4 -0
  4. train.py +0 -1
.gitignore CHANGED
@@ -1,2 +1 @@
1
- data/*
2
- model/*
 
1
+ data/*
 
model/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74e647e42e188420cbe122cb654606dccb566a19401a4477cdd19706cfc0e921
3
+ size 319936
requirments.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ torch
2
+ torchvision
3
+ gradio
4
+ PIL
train.py CHANGED
@@ -2,7 +2,6 @@ import torch
2
  import torch.nn as nn
3
  import torchvision
4
  import torchvision.transforms as transforms
5
- import matplotlib.pyplot as plt
6
  from NeuralNet import NeuralNet
7
 
8
  # Device Config
 
2
  import torch.nn as nn
3
  import torchvision
4
  import torchvision.transforms as transforms
 
5
  from NeuralNet import NeuralNet
6
 
7
  # Device Config