Update app.py

app.py

@@ -1,17 +1,15 @@
-
 import os
-import numpy as np
 from flask import Flask, request, send_file, jsonify
 from PIL import Image
 import torch
 from io import BytesIO
-import requests
-from werkzeug.utils import secure_filename
+from huggingface_hub import hf_hub_download
 
 app = Flask(__name__)
 
-# تحميل موديل U^2-Net (نموذج إزالة الخلفية)
-model = torch.hub.load('opencv/opencv_contrib', 'u2net', pretrained=True)
+# تحميل نموذج U^2-Net باستخدام Hugging Face Hub
+MODEL_PATH = hf_hub_download("xuebinqin/U-2-Net", "u2net.pth")
+model = torch.load(MODEL_PATH)
 model.eval()
 
 # مسار لحفظ الصور المرفوعة
@@ -23,7 +21,6 @@ app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 def remove_background(image_path):
     image = Image.open(image_path).convert("RGB")
     # معالجة الصورة باستخدام نموذج U²-Net (التفاصيل في الكود)
-    # هنا نستخدم OpenCV للتعامل مع الصورة
     return image  # هذا هو النص الاختياري لدمج إزالة الخلفية
 
 # دالة لإضافة خلفية بيضاء