Update app.py

app.py

@@ -1,18 +1,31 @@
-import gradio as gr
-from transformers import AutoImageProcessor, SapiensForImageSegmentation
+import torch
+from transformers import AutoImageProcessor
 from PIL import Image
+import requests
+from io import BytesIO
 
+# آدرس مدل
+model_url = "https://huggingface.co/facebook/sapiens/resolve/main/sapiens_lite_host/torchscript/normal/checkpoints/sapiens_0.3b"
+model = torch.jit.load(model_url)
 
-model_name = "facebook/sapiens"
-processor = AutoImageProcessor.from_pretrained(model_name)
-model = SapiensForImageSegmentation.from_pretrained(model_name)
+# بارگذاری پردازشگر تصویر
+processor = AutoImageProcessor.from_pretrained("facebook/sapiens")
 
 def segment_image(image):
+    # پردازش تصویر
     inputs = processor(images=image, return_tensors="pt")
-    outputs = model(**inputs)
-    segmentation = outputs.logits.argmax(dim=1).detach().cpu().numpy()[0]
+    
+    # اجرای مدل روی تصویر پردازش شده
+    with torch.no_grad():
+        outputs = model(inputs['pixel_values'])
+    
+    # فرض می‌کنیم خروجی یک ماسک است
+    segmentation = outputs.argmax(dim=1).detach().cpu().numpy()[0]
+    
     return Image.fromarray(segmentation)
 
+# رابط Gradio
+import gradio as gr
 interface = gr.Interface(
     fn=segment_image,
     inputs=gr.Image(type="pil"),