Update app.py

app.py

@@ -29,8 +29,9 @@ def load_image_into_numpy_array(path):
     return pil_image_as_numpy_array(image)            
 
 def load_model():
-    model_dir = 'saved_model'    
-    detection_model = tf.saved_model.load(str(model_dir))
+    download_dir = snapshot_download(REPO_ID)
+    saved_model_dir = os.path.join(download_dir, "saved_model")
+    detection_model = tf.saved_model.load(saved_model_dir)
     return detection_model
 
 # samples_folder = 'test_samples