From 00e6b4a164bddafac24f00c10b0e5cb4b1afce6f Mon Sep 17 00:00:00 2001
From: Nataniel Ruiz <nruiz@Nataniels-MacBook-Pro.local>
Date: Wed, 25 Dec 2019 20:25:08 -0400
Subject: [PATCH] next

---
 cyclegan/models/test_model.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cyclegan/models/test_model.py b/cyclegan/models/test_model.py
index 39f85c1..7b5aaea 100644
--- a/cyclegan/models/test_model.py
+++ b/cyclegan/models/test_model.py
@@ -1,7 +1,9 @@
 from .base_model import BaseModel
 from . import networks
 from util import attacks
-
+import numpy as np
+import torch
+import torch.nn.functional as F
 
 class TestModel(BaseModel):
     """ This TesteModel can be used to generate CycleGAN results for only one direction.