remove old train/predict script

Kawaeee · Feb 10, 2022 · 42a4288 · 42a4288
1 parent 625b84d
commit 42a4288
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 391 deletions.
diff --git a/predictor.ipynb b/predictor.ipynb
diff --git a/requirements.txt b/requirements.txt
@@ -8,6 +8,6 @@ streamlit==1.5.0
 # [STREAMLIT-DEPLOYMENT-GUIDE] Add "+cpu" on torch and torchvision to get CPU inference support
 # NOTE: On macOS system, it might be unable to install with +cpu flag
 --find-links https://download.pytorch.org/whl/torch_stable.html
-torch==1.8.1
-torchvision==0.9.1
+torch==1.8.1+cpu
+torchvision==0.9.1+cpu
 tqdm==4.60.0
diff --git a/train.py b/train.py