Update app.py

app.py

@@ -8,17 +8,13 @@ from PIL import Image
 import numpy as np
 import cv2
 from Utils import *
-import tensorflow as tf
-from huggingface_hub import hf_hub_download
 
 
+from huggingface_hub import from_pretrained_keras
 
-file_path = hf_hub_download("SerdarHelli/Knee-View-Merchant-Landmark-Detection","keras",
-                        use_auth_token=os.environ.get("TOKEN_FROM_SECRET"))
+model = from_pretrained_keras("SerdarHelli/Knee-View-Merchant-Landmark-Detection",use_auth_token=os.environ.get("TOKEN_FROM_SECRET"))
 
 
-model=tf.keras.load_model(file_path)
-
 st.subheader("Upload Merchant Knee View")
 image_file = st.file_uploader("Upload Images", type=["dcm"])