Hugging Face's logo

Spaces:
scbirlab
/
lchemme
Runtime error

App Files Community
Eachan Johnson commited on
Commit
cd9c1a6
·
1 Parent(s): 9e1ab04

Don't use non-default model cache

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,8 +29,8 @@ MODELS = (
29
  )
30
 
31
  models = {model_name: (
32
- AutoTokenizer.from_pretrained(model_name, cache_dir="model-cache"),
33
- AutoModelForSeq2SeqLM.from_pretrained(model_name, cache_dir="model-cache"),
34
  ) for model_name in MODELS}
35
 
36
  def load_input_data(file: TextIOWrapper) -> pd.DataFrame:
 
29
  )
30
 
31
  models = {model_name: (
32
+ AutoTokenizer.from_pretrained(model_name),
33
+ AutoModelForSeq2SeqLM.from_pretrained(model_name),
34
  ) for model_name in MODELS}
35
 
36
  def load_input_data(file: TextIOWrapper) -> pd.DataFrame: