Update app.py

app.py

@@ -8,7 +8,6 @@ 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'
@@ -30,10 +29,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":
-            try:
-                return from_pretrained_keras("SerdarHelli/Segmentation-of-Teeth-in-Panoramic-X-ray-Image-Using-U-Net")
-            except:
-                return tf.keras.models.load_model(model_path)
+            return load_model(model_path)
         elif model_name == "Caries Detection":
             return YOLO(model_path)
         else: