Ruian7P commited on
Commit
e6cc031
·
1 Parent(s): ebaa8bd

test emuru

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -37,8 +37,9 @@ def load_model(model_name="imuru_t5_small"):
37
  from transformers import AutoModel
38
 
39
  model = AutoModel.from_pretrained(
40
- model_zoo[model_name]["repo_id"],
41
- subfolder=model_zoo[model_name]["model_name"],
 
42
  trust_remote_code=True
43
  )
44
  model.eval()
 
37
  from transformers import AutoModel
38
 
39
  model = AutoModel.from_pretrained(
40
+ # model_zoo[model_name]["repo_id"],
41
+ # subfolder=model_zoo[model_name]["model_name"],
42
+ "blowing-up-groundhogs/emuru"
43
  trust_remote_code=True
44
  )
45
  model.eval()