Update app.py

app.py

@@ -8,6 +8,9 @@ import shinyswatch
 
 from shiny import App, Inputs, Outputs, Session, reactive, render, req, ui
 
+import os
+os.environ["TRANSFORMERS_CACHE"] = Path(__file__).parent.resolve() / "cache"
+
 from transformers import SamModel, SamConfig, SamProcessor
 import torch
 
@@ -80,8 +83,6 @@ def server(input: Inputs, output: Outputs, session: Session):
     @render.image
     def uploaded_image():
         """Displays the uploaded image"""
-        from pathlib import Path
-
         img_src = uploaded_image_path()
         if img_src:
             dir = Path(__file__).resolve().parent