Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 import cv2
-import gdown
 import gradio as gr
 import tensorflow as tf
 import urllib.request
@@ -15,14 +14,14 @@ resized_shape = (768, 768, 3)
 IMG_SCALING = (1, 1)
 
 
-# Download the model file
-def download_model():
-    url = "https://drive.google.com/uc?id=1FhICkeGn6GcNXWTDn1s83ctC-6Mo1UXk"
-    output = "seg_unet_model.h5"
-    gdown.download(url, output, quiet=False)
-    return output
+# # Download the model file
+# def download_model():
+#     url = "https://drive.google.com/uc?id=1FhICkeGn6GcNXWTDn1s83ctC-6Mo1UXk"
+#     output = "seg_unet_model.h5"
+#     gdown.download(url, output, quiet=False)
+#     return output
 
-model_file = download_model()
+model_file = "./seg_unet_model.h5"
 
 #Custom objects for model