add more test + structure

Mirtia · Jul 10, 2023 · f62d826 · f62d826
1 parent a2613e7
commit f62d826
Show file tree

Hide file tree

Showing 11 changed files with 12 additions and 2 deletions.
diff --git a/input/wikiart_input b/input/wikiart_input
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,10 @@
+aiohttp==3.8.4
+beautifulsoup4==4.10.0
+Google_Images_Search==1.4.6
+imageai==3.0.3
+Pillow==9.5.0
+python-dotenv==1.0.0
+RealESRGAN==1.0
+RealESRGAN==0.3.0
+Requests==2.31.0
+torch==2.0.1+cu118
diff --git a/crawler.py → src/crawler.py b/crawler.py → src/crawler.py
diff --git a/google_crawler.py → src/google_crawler.py b/google_crawler.py → src/google_crawler.py
diff --git a/image_upscaler.py → src/image_upscaler.py b/image_upscaler.py → src/image_upscaler.py
diff --git a/main.py → src/main.py b/main.py → src/main.py
@@ -11,8 +11,8 @@ def main():
  parser.add_argument("-c", "--crawl", action="store_true", help="Enable crawling")
  parser.add_argument("-t", "--train", action="store_true", help="Train model")
  parser.add_argument("-i", "--input", dest="input", help="Specify input image to test model")
- parser.add_argument("-f", "--file", dest="file", help="Specify input file", required=True)
- parser.add_argument("-o", "--output", dest="output", help="Specify output file", required=True)
+ parser.add_argument("-f", "--file", dest="file", help="Specify input file")
+ parser.add_argument("-o", "--output", dest="output", help="Specify output file")
  args = parser.parse_args()
 
  if args.crawl:

diff --git a/model.py → src/model.py b/model.py → src/model.py
diff --git a/wikimedia_crawler.py → src/wikimedia_crawler.py b/wikimedia_crawler.py → src/wikimedia_crawler.py
diff --git a/tests/bacon_expressionism_cubism_surrealism.jpg b/tests/bacon_expressionism_cubism_surrealism.jpg
diff --git a/tests/my_artwork.png b/tests/my_artwork.png
diff --git a/tests/waterhouse_pre-rphaelite.jpg b/tests/waterhouse_pre-rphaelite.jpg