Upload app.py with huggingface_hub

app.py

@@ -48,14 +48,16 @@ def load_model(model_name: str):
     model_config = MODELS_SPECIFIC_CONFIGS.get(model_name, {})
     model = build_interfuser_model(model_config)
     if not os.path.exists(weights_path):
-        gr.Warning(f"ملف الأوزان '{weights_path}' غير موجود.")
+        # استبدال gr.Warning برمي استثناء خطأ
+        raise gr.Error(f"ملف الأوزان '{weights_path}' غير موجود.")
     else:
         try:
             state_dic = torch.load(weights_path, map_location=device, weights_only=True)
             model.load_state_dict(state_dic)
             print(f"تم تحميل أوزان النموذج '{model_name}' بنجاح.")
         except Exception as e:
-            gr.Warning(f"فشل تحميل الأوزان للنموذج '{model_name}': {e}.")
+            # استبدال gr.Warning برمي استثناء خطأ
+            raise gr.Error(f"فشل تحميل الأوزان للنموذج '{model_name}': {e}.")
     model.to(device)
     model.eval()
     return model, f"تم تحميل نموذج: {model_name}"
@@ -232,5 +234,5 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"), cs
 # ==============================================================================
 if __name__ == "__main__":
     if not available_models:
-        print("تحذير: لم يتم العثور على أي ملفات نماذج (.pth) في مجلد 'model/weights'.")
-    demo.queue().launch(debug=True, share=True) # share=True لإنشاء رابط عام مؤقت
+        print("تحذير: لم يتم العثور على أي ملفات نماذج (.pth) في مجلد 'model'.")
+    demo.queue().launch(debug=True, share=True) # share=True لإنشاء رابط عام مؤقت