Update app.py

app.py

@@ -8,6 +8,8 @@ from tensorflow.keras.models import load_model
 from PIL import Image, ImageDraw, ImageFont
 from ultralytics import YOLO
 import cv2
+from huggingface_hub import from_pretrained_keras
+
 
 class Config:
     ASSETS_DIR = './assets'
@@ -29,7 +31,7 @@ class ModelManager:
     def load_model(model_name: str):
         model_path = os.path.join(Config.MODELS_DIR, Config.MODELS[model_name])
         if model_name == "Dental X-Ray Segmentation":
-            return tf.keras.models.load_model(model_path)
+            return from_pretrained_keras("SerdarHelli/Segmentation-of-Teeth-in-Panoramic-X-ray-Image-Using-U-Net")
         elif model_name == "Caries Detection":
             return YOLO(model_path)
         else: