nico-x commited on
Commit
eced394
·
1 Parent(s): 3ed1134

addedd app.py and model for hf space

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. app.py +8 -0
.gitignore CHANGED
@@ -14,6 +14,7 @@ venv/
14
 
15
  # PyTorch checkpoints
16
  *.pt
 
17
 
18
  # Gradio session files
19
  gradio_cached_examples/
 
14
 
15
  # PyTorch checkpoints
16
  *.pt
17
+ ! checkpoints/transformer_mnist.pt # allow the model to be pushed to HF Space
18
 
19
  # Gradio session files
20
  gradio_cached_examples/
app.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ # entrpoint for HuggingFace Space
2
+
3
+ import sys
4
+ sys.path.append('.')
5
+
6
+ from app.gradio_app import demo
7
+
8
+ demo.launch()