Ariamehr commited on
Commit
de64d08
1 Parent(s): e5f4670

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ import requests
5
  import gradio as gr
6
 
7
  # لود پردازشگر تصویر و مدل
8
- processor = AutoImageProcessor.from_pretrained("facebook/sapiens")
9
- model = AutoModelForImageSegmentation.from_pretrained("facebook/sapiens")
10
 
11
  # تابع برای پردازش تصویر ورودی و اعمال مدل بر روی آن
12
  def segment_image(image):
 
5
  import gradio as gr
6
 
7
  # لود پردازشگر تصویر و مدل
8
+ processor = AutoImageProcessor.from_pretrained("facebook/sapiens/tree/main/sapiens_lite_host/torchscript/seg/checkpoints/sapiens_0.3b")
9
+ model = AutoModelForImageSegmentation.from_pretrained("facebook/sapiens/tree/main/sapiens_lite_host/torchscript/seg/checkpoints/sapiens_0.3b")
10
 
11
  # تابع برای پردازش تصویر ورودی و اعمال مدل بر روی آن
12
  def segment_image(image):