Coots commited on
Commit
befba3b
·
verified ·
1 Parent(s): 7d316b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ os.makedirs(cache_dir, exist_ok=True)
13
  hf_token = os.getenv("HF_TOKEN")
14
 
15
  # Load model and tokenizer
16
- model_id = "HuggingFaceH4/zephyr-7b-beta"
17
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token, cache_dir=cache_dir)
18
  model = AutoModelForCausalLM.from_pretrained(
19
  model_id,
 
13
  hf_token = os.getenv("HF_TOKEN")
14
 
15
  # Load model and tokenizer
16
+ model_id = "google/flan-t5-large"
17
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token, cache_dir=cache_dir)
18
  model = AutoModelForCausalLM.from_pretrained(
19
  model_id,