diff --git a/pages/Image_prediction.py b/pages/Image_prediction.py
index 122a546..fccdd65 100644
--- a/pages/Image_prediction.py
+++ b/pages/Image_prediction.py
@@ -10,6 +10,7 @@
 import random
 import os
 import tempfile
+import gdown
 
 # Model Links
 MODEL_LINKS = {