Upload folder using huggingface_hub

app.py

@@ -16,8 +16,6 @@ else: device = 'cpu'
 print('DEVICE:', device)
 if device == 'cuda': print('CUDA DEVICE:', torch.cuda.get_device_name())
 
-torch.manual_seed(42)
-
 model = VGG_19().to(device).eval()
 for param in model.parameters():
     param.requires_grad = False

inference.py

@@ -26,8 +26,8 @@ def inference(
     content_image,
     style_features,
     lr,
-    iterations=3,
-    optim_caller=optim.LBFGS,
+    iterations=35,
+    optim_caller=optim.AdamW,
     alpha=1,
     beta=1
 ):