Update app.py

app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-from loadimg import load_img
 from transformers import AutoModelForImageSegmentation
 import torch
 from torchvision import transforms
@@ -7,12 +6,16 @@ from PIL import Image
 import requests
 from io import BytesIO
 
-# إعداد الجهاز
+# إعداد الجهاز للعمل على CPU
+device = torch.device("cpu")
 torch.set_float32_matmul_precision("high")
+
+# تحميل النموذج
 birefnet = AutoModelForImageSegmentation.from_pretrained(
     "ZhengPeng7/BiRefNet", trust_remote_code=True
-)
+).to(device)
 
+# تحويل الصورة
 transform_image = transforms.Compose(
     [
         transforms.Resize((1024, 1024)),
@@ -24,7 +27,7 @@ transform_image = transforms.Compose(
 # دالة المعالجة
 def process(image):
     image_size = image.size
-    input_images = transform_image(image).unsqueeze(0).to("cuda")
+    input_images = transform_image(image).unsqueeze(0).to(device)
     with torch.no_grad():
         preds = birefnet(input_images)[-1].sigmoid().cpu()
     pred = preds[0].squeeze()