fix import error

goodday-g1 · Nov 8, 2018 · 5dcdf77 · 5dcdf77
1 parent 1f2df8f
commit 5dcdf77
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/colab_demo/faceswap-GAN_colab_demo.ipynb b/colab_demo/faceswap-GAN_colab_demo.ipynb
@@ -589,7 +589,7 @@
     "# VGGFace ResNet50\n",
     "#vggface = VGGFace(include_top=False, model='resnet50', input_shape=(224, 224, 3))'\n",
     "\n",
-    "from .vggface_models import RESNET50\n",
+    "from colab_demo.vggface_models import RESNET50\n",
     "vggface = RESNET50(include_top=False, weights=None, input_shape=(224, 224, 3))\n",
     "vggface.load_weights(\"rcmalli_vggface_tf_notop_resnet50.h5\")\n",
     "\n",