Update app.py

app.py

@@ -8,12 +8,11 @@ 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 = os.path.join(os.path.dirname(__file__), 'assets')
-    MODELS_DIR = os.path.join(ASSETS_DIR, 'models')
-    FONT_DIR = os.path.join(ASSETS_DIR, 'arial.ttf')
+    ASSETS_DIR = './assets'
+    MODELS_DIR = './models'
+    FONT_DIR = './assets/arial.ttf'
     MODELS = {
         "Calculus and Caries Classification": "classification.h5",
         "Caries Detection": "detection.pt",
@@ -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 tf.keras.models.load_model(model_path)
         elif model_name == "Caries Detection":
             return YOLO(model_path)
         else:
@@ -256,6 +252,16 @@ class GradioInterface:
         <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/css/bulma.min.css">
         <link rel="stylesheet" href="https://use.fontawesome.com/releases/v5.15.4/css/all.css">
         {app_styles}
+        <div class="app-header">
+            <h1 class="app-title">AI Dentistry Application</h1>
+            <h2 class="app-subtitle">Model Selection and Image Processing</h2>
+            <p class="app-description">
+                This application allows you to select different models for dental image processing tasks.
+            </p>
+        </div>
+        """
+
+        header_html = f"""
         <div class="app-header">
             <h1 class="app-title">AI in Dentistry</h1>
             <h2 class="app-subtitle"> Advancing Imaging and Clinical Transcription</h2>
@@ -265,6 +271,16 @@ class GradioInterface:
         </div>
         """
 
+        js_func = """
+        function refresh() {
+            const url = new URL(window.location);
+            if (url.searchParams.get('__theme') !== 'dark') {
+                url.searchParams.set('__theme', 'dark');
+                window.location.href = url.href;
+            }
+        }
+        """
+
         def process_image(image, model_name):
             result = self.image_processor.process_image(image, model_name)
             return result
@@ -273,7 +289,7 @@ class GradioInterface:
             examples = self.preloaded_examples[model_name]
             return gr.Dataset(samples=[[example] for example in examples])
 
-        with gr.Blocks() as demo:
+        with gr.Blocks(js=js_func, theme=gr.themes.Default()):
             gr.HTML(header_html)
             with gr.Row(elem_classes="content-container"):
                 with gr.Column():
@@ -312,4 +328,5 @@ def main():
     demo.launch(share=False)
 
 if __name__ == "__main__":
-    main()
+    main()
+