Adjusting requirements

app.py

@@ -13,7 +13,6 @@ import time
 import requests
 sys.path.append('./CLIP')
 sys.path.append('deep-image-prior')
-import cv2
 from einops import rearrange
 import gc
 import imageio
@@ -42,6 +41,8 @@ clip_normalize = T.Normalize(mean=[0.48145466, 0.4578275, 0.40821073], std=[0.26
 clip_model = 'ViT-B/16'
 sideX, sideY = 512, 512  # Resolution
 inv_color_scale = 1.6
+anneal_lr = True
+display_augs = False 
 
 class MakeCutouts(torch.nn.Module):
     def __init__(self, cut_size, cutn):

requirements.txt

@@ -8,5 +8,4 @@ lpips
 numpy
 imageio
 einops
-madgrad
-cv2
+madgrad