Update app.py

app.py

@@ -5,7 +5,7 @@ import torch.nn as nn
 from torchvision import transforms
 from PIL import Image
 import numpy as np
-import tensorflow as tf
+from tensorflow.keras.models import load_model
 import os
 import requests
 import cv2
@@ -16,7 +16,7 @@ st.markdown("Upload a full-face image. The system will detect the affected side
 @st.cache_resource
 def download_models():
     model_urls = {
-        "cnn_stroke_model.h5": "https://huggingface.co/AdhamQQ/cnn_stroke_model/resolve/main/cnn_stroke_model.h5",
+        "cnn_stroke_model": "https://huggingface.co/AdhamQQ/cnn_stroke_model/resolve/main/cnn_stroke_model",
         "left_side_pain_classifier.pth": "https://huggingface.co/AdhamQQ/cnn_stroke_model/resolve/main/left_side_pain_classifier.pth",
         "right_side_pain_classifier.pth": "https://huggingface.co/AdhamQQ/cnn_stroke_model/resolve/main/right_side_pain_classifier.pth",
     }
@@ -40,7 +40,7 @@ def download_models():
             f.write(r.content)
 
     # Load stroke model
-    stroke_model = tf.keras.models.load_model("cnn_stroke_model.h5")
+    stroke_model = load_model("cnn_stroke_model")
 
     # Define corrected PainModel class
     class PainModel(nn.Module):