working finally

Dockerfile

@@ -1,15 +1,10 @@
-FROM python:3.9
+FROM python:3.11
 
 WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-
-RUN apt-get update && \
-    apt-get install -y git-lfs && \
-    rm -rf /var/lib/apt/lists/* && \
-    git lfs install
+RUN pip install -r ./requirements.txt
 
 COPY . .
 
@@ -19,6 +14,5 @@ USER root
 RUN mkdir -p /.cache
 RUN chmod 777 /.cache
 RUN chmod 777 /code/*
-USER user
 
 CMD ["shiny", "run", "app.py", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -32,9 +32,9 @@ def server(input: Inputs, output: Outputs, session: Session):
             return None
         
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-        model2 = AutoModel.from_pretrained("ansal/sidewalk-segment")
+        model2 = AutoModel.from_pretrained("ansal/sidewalk-segment", device_map = 'cpu')
         model2.eval()
-        model2.to(device)
+        model2.to('cpu')
 
         image = Image.open(file[0]["datapath"]).convert('RGB')
         transform = image_resize_transform
@@ -47,7 +47,10 @@ def server(input: Inputs, output: Outputs, session: Session):
         
         mask_tensor = predicted_masks.cpu().detach().squeeze()
         mask_array = mask_tensor.numpy()  
-        mask_array = (mask_array * 255).astype(np.uint8)  
+        mask_array = np.nan_to_num(mask_array)
+        mask_array = mask_array * 255
+        mask_array = np.clip(mask_array, 0, 255)  
+        mask_array = mask_array.astype(np.uint8) 
         mask = Image.fromarray(mask_array)
         mask = mask.resize((1024, 1024), Image.LANCZOS)
         mask = mask.convert('RGBA')

model.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:41049da640c5d9c033ea78378413bbf9f299bd17f989c8ebd126b7a66b5aa1e1
-size 375067302

requirements.txt

@@ -1,11 +1,12 @@
-shiny 
-shinywidgets
-ipyleaflet
-geopy
-faicons
-requests
-Pillow
-torch
-numpy
+shiny==0.9.0
+shinywidgets==0.3.2
+ipyleaflet==0.19.0
+geopy==2.4.1
+faicons==0.2.2
+requests==2.31.0
+Pillow==10.0.0
+torch==2.1.1
+numpy==1.24.3
 transformers
-torchvision
+torchvision==0.16.1
+accelerate==0.30.0