Update app.py

app.py

@@ -9,12 +9,22 @@ import numpy as np
 import cv2
 from Utils import *
 
+from huggingface_hub import hf_hub_download,from_pretrained_keras
+
+import tensorflow as tf
 
-from huggingface_hub import from_pretrained_keras
 
 model = from_pretrained_keras("SerdarHelli/Knee-View-Merchant-Landmark-Detection",use_auth_token=os.environ.get("TOKEN_FROM_SECRET"))
 
 
+
+file_path = hf_hub_download("SerdarHelli/Knee-View-Merchant-Landmark-Detection","keras",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"])