Spaces:

m-ric
/

beam_search_visualizer

Running on Zero

m-ric HF staff commited on Mar 21

Commit

f3f39fc

•

1 Parent(s): e774464

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -114,8 +114,8 @@ STYLE = """
 from transformers import GPT2Tokenizer, AutoModelForCausalLM, AutoTokenizer
 import numpy as np
-tokenizer = AutoTokenizer.from_pretrained("Locutusque/TinyMistral-248M-v2")
-model = AutoModelForCausalLM.from_pretrained("Locutusque/TinyMistral-248M-v2")
 tokenizer.pad_token_id = tokenizer.eos_token_id
 print("Loading finished.")
 def generate_html(token, node):

 from transformers import GPT2Tokenizer, AutoModelForCausalLM, AutoTokenizer
 import numpy as np
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
+model = AutoModelForCausalLM.from_pretrained("gpt2")
 tokenizer.pad_token_id = tokenizer.eos_token_id
 print("Loading finished.")
 def generate_html(token, node):