feat: no grad

app.py

@@ -16,6 +16,7 @@ def flush():
     torch.cuda.empty_cache()
 
 
+@torch.no_grad()
 def extract(image):
     extractor = MangaLineExtractor("./models/erika.pth", "cpu")
     result = extractor.predict(image)
@@ -24,6 +25,7 @@ def extract(image):
     return result
 
 
+@torch.no_grad()
 def convert_to_sketch(image):
     to_sketch = Anime2Sketch("./models/netG.pth", "cpu")
     result = to_sketch.predict(image)