Trang Dang commited on
Commit
92badd3
1 Parent(s): 7008c4a

change environ

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. run.py +0 -2
app.py CHANGED
@@ -4,6 +4,8 @@ import pandas as pd
4
  import seaborn as sns
5
  import shinyswatch
6
  import run
 
 
7
  import matplotlib.pyplot as plt
8
 
9
  from shiny import App, Inputs, Outputs, Session, reactive, render, req, ui
 
4
  import seaborn as sns
5
  import shinyswatch
6
  import run
7
+ import os
8
+ os.environ['MPLCONFIGDIR'] = "/code/configs"
9
  import matplotlib.pyplot as plt
10
 
11
  from shiny import App, Inputs, Outputs, Session, reactive, render, req, ui
run.py CHANGED
@@ -2,13 +2,11 @@ from transformers import SamModel, SamConfig, SamProcessor
2
  import torch
3
  import numpy as np
4
  import app
5
- import os
6
  from PIL import Image
7
 
8
  def pred(src):
9
  # Load the model configuration
10
  cache_dir = "/code/cache"
11
-
12
  model_config = SamConfig.from_pretrained("facebook/sam-vit-base", cache_dir=cache_dir)
13
  processor = SamProcessor.from_pretrained("facebook/sam-vit-base", cache_dir=cache_dir)
14
 
 
2
  import torch
3
  import numpy as np
4
  import app
 
5
  from PIL import Image
6
 
7
  def pred(src):
8
  # Load the model configuration
9
  cache_dir = "/code/cache"
 
10
  model_config = SamConfig.from_pretrained("facebook/sam-vit-base", cache_dir=cache_dir)
11
  processor = SamProcessor.from_pretrained("facebook/sam-vit-base", cache_dir=cache_dir)
12