Update app.py

app.py

@@ -98,25 +98,25 @@ class DFUTissueSegNet(nn.Module):
 # تحميل النموذج
 # ======================================================
 def initialize_model():
-    """تحميل DFUTissueSegNet"""
+    """تحميل DFUTissueSegNet من Google Drive"""
     global segmenter
-    MODEL_URL = "https://github.com/JoshKowi/DFUTissueSegNet/raw/main/Models/DFUTissueSegNet_best.pth"
-    MODEL_PATH = "DFUTissueSegNet_best.pth"
+    MODEL_URL = "https://drive.google.com/uc?id=1Ovaczsjdp3E-_gYF2pbUibDjPWAC1a6c"
+    MODEL_PATH = "best_model_5.pth"
 
     if not os.path.exists(MODEL_PATH):
-        print("📥 تحميل النموذج من GitHub...")
+        print("📥 تحميل DFUTissueSegNet من Google Drive...")
         gdown.download(MODEL_URL, MODEL_PATH, quiet=False)
 
     try:
-        print("🔄 تحميل نموذج DFUTissueSegNet...")
+        print("🔄 تحميل النموذج...")
         segmenter = DFUTissueSegNet(num_classes=3)
         state_dict = torch.load(MODEL_PATH, map_location=DEVICE)
         segmenter.load_state_dict(state_dict)
         segmenter.to(DEVICE)
         segmenter.eval()
-        print("✅ تم تحميل DFUTissueSegNet بنجاح.")
+        print("✅ تم تحميل DFUTissueSegNet من Google Drive بنجاح.")
     except Exception as e:
-        print(f"❌ فشل تحميل النموذج: {e}")
+        print(f"❌ خطأ في تحميل النموذج: {e}")
         segmenter = None