Update app.py

app.py

@@ -5,6 +5,7 @@ from PIL import Image
 import torch
 import numpy as np
 import torchvision.transforms as transforms
+from transformers import AutoModel
 from shiny import App, Inputs, Outputs, Session, reactive, render, ui
 from shiny.types import FileInfo
 import base64
@@ -31,7 +32,7 @@ def server(input: Inputs, output: Outputs, session: Session):
             return None
         
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-        model2 = SamModel.from_pretrained("facebook/sam-vit-base")
+        model2 = AutoModel.from_pretrained("ansal/sidewalk-segment")
         model2.load_state_dict(torch.load('model.pth', map_location=device))
         model2.eval()
         model2.to(device)