Spaces:
Build error
Build error
Trang Dang
commited on
Commit
•
92badd3
1
Parent(s):
7008c4a
change environ
Browse files
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 |
|