aagoluoglu commited on
Commit
b8d0481
·
verified ·
1 Parent(s): 9ae3b6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -9,8 +9,6 @@ import shinyswatch
9
  from shiny import App, Inputs, Outputs, Session, reactive, render, req, ui
10
 
11
  import os
12
- os.environ["TRANSFORMERS_CACHE"] = str(Path(__file__).parent.resolve() / "hf_cache")
13
-
14
  from transformers import SamModel, SamConfig, SamProcessor
15
  import torch
16
 
@@ -61,6 +59,7 @@ app_ui = ui.page_fillable(
61
 
62
 
63
  def server(input: Inputs, output: Outputs, session: Session):
 
64
  # Load the model configuration
65
  model_config = SamConfig.from_pretrained("facebook/sam-vit-base")
66
  processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
 
9
  from shiny import App, Inputs, Outputs, Session, reactive, render, req, ui
10
 
11
  import os
 
 
12
  from transformers import SamModel, SamConfig, SamProcessor
13
  import torch
14
 
 
59
 
60
 
61
  def server(input: Inputs, output: Outputs, session: Session):
62
+ os.environ["TRANSFORMERS_CACHE"] = "./hf_cache"
63
  # Load the model configuration
64
  model_config = SamConfig.from_pretrained("facebook/sam-vit-base")
65
  processor = SamProcessor.from_pretrained("facebook/sam-vit-base")