Spaces:

aagoluoglu
/

SAMSidewalksDemo

Runtime error

aagoluoglu commited on Apr 25, 2024

Commit

3a5c18f

verified ·

1 Parent(s): 9507cd3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,8 +8,8 @@ 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

 from shiny import App, Inputs, Outputs, Session, reactive, render, req, ui
+import os
+os.environ["TRANSFORMERS_CACHE"] = Path(__file__).parent.resolve() / "hf_cache"
 from transformers import SamModel, SamConfig, SamProcessor
 import torch