Update app.py

app.py

@@ -8,6 +8,9 @@ import shinyswatch
 
 from shiny import App, Inputs, Outputs, Session, reactive, render, req, ui
 
+from transformers import SamModel, SamConfig, SamProcessor
+import torch
+
 sns.set_theme()
 
 www_dir = Path(__file__).parent.resolve() / "www"
@@ -55,6 +58,17 @@ app_ui = ui.page_fillable(
 
 
 def server(input: Inputs, output: Outputs, session: Session):
+    # Load the model configuration
+    model_config = SamConfig.from_pretrained("facebook/sam-vit-base")
+    processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
+    
+    # Create an instance of the model from my fine-tuned model with the loaded configuration
+    model = SamModel.from_pretrained("aagoluoglu/SAM_Sidewalks", config=model_config)
+    
+    # set the device to cuda if available, otherwise use cpu
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    model.to(device)
+    
     @reactive.Calc
     def uploaded_image_path() -> str:
         """Returns the path to the uploaded image"""