From 8eefddc0c8b270c4ed5b43d8b0d4f0789e04d157 Mon Sep 17 00:00:00 2001
From: onb1259 <onb1259@onb.ac.at>
Date: Mon, 14 Aug 2023 14:33:16 +0200
Subject: [PATCH] added requirements file

---
 .gitignore            | 1 +
 requirements.txt      | 6 ++++++
 train_colorizer.ipynb | 2 +-
 3 files changed, 8 insertions(+), 1 deletion(-)
 create mode 100644 requirements.txt

diff --git a/.gitignore b/.gitignore
index 0843976..7368bbb 100644
--- a/.gitignore
+++ b/.gitignore
@@ -4,3 +4,4 @@ __pycache__
 *.png
 .ipynb_checkpoints
 postcards_256/*
+venv
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..97bf5f5
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,6 @@
+torch
+scikit-image
+matplotlib
+numpy
+pillow
+requests
diff --git a/train_colorizer.ipynb b/train_colorizer.ipynb
index 95662f1..d2bc462 100644
--- a/train_colorizer.ipynb
+++ b/train_colorizer.ipynb
@@ -17,7 +17,6 @@
    "source": [
     "import os\n",
     "import requests\n",
-    "import torch\n",
     "from torch.utils.data import random_split\n",
     "from torch.utils.data import DataLoader\n",
     "from models.ColDataset import ColDataset\n",
@@ -171,6 +170,7 @@
    ],
    "source": [
     "# Get a random AKON postcard\n",
+    "import torch\n",
     "from random import randint\n",
     "from skimage import io\n",
     "from iiif_utils import get_manifesturls_from_collectionurl, get_imgurls_from_manifesturl\n",
-- 
GitLab