christophschuhmann / improved-aesthetic-predictor

CLIP+MLP Aesthetic Score Predictor
Apache License 2.0
897 stars 88 forks source link

Added requirements.txt and CPU only inference option #10

Open varna9000 opened 1 year ago

varna9000 commented 1 year ago
  1. Created requirements.txt for easier pip install
  2. Added checks for CPU/CUDA in simple_inference.py, so we run the model also on CPU only systems. Tested on Macbook Pro M1 with python 3.11