inclusive-ml commited on
Commit
26fb881
1 Parent(s): 1734a82

translate, summarize, similarity

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def load_model_cache():
19
  if not os.path.exists(CACHE_DIR):
20
  os.mkdir(CACHE_DIR)
21
  # download the files from huggingface repo and load the model with tensorflow
22
- snapshot_download(repo_id="widged/translate-to-french", cache_dir=CACHE_DIR)
23
  saved_model_path = os.path.join(CACHE_DIR, os.listdir(CACHE_DIR)[0])
24
  return saved_model_path
25
  def dashboard(model):
 
19
  if not os.path.exists(CACHE_DIR):
20
  os.mkdir(CACHE_DIR)
21
  # download the files from huggingface repo and load the model with tensorflow
22
+ snapshot_download(repo_id="stevekola/T5", cache_dir=CACHE_DIR)
23
  saved_model_path = os.path.join(CACHE_DIR, os.listdir(CACHE_DIR)[0])
24
  return saved_model_path
25
  def dashboard(model):