decoding_visualizer

Running on T4

Jofthomas commited on Jan 23

Commit

69cd7c3

verified ·

1 Parent(s): 7a92631

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ import numpy as np
 import gradio as gr
 import spaces
-tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b-it")
-model = AutoModelForCausalLM.from_pretrained("google/gemma-2-2b-it")
 print("Loading finished.")

 import gradio as gr
 import spaces
+tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM2-360M-Instruct")
+model = AutoModelForCausalLM.from_pretrained("HuggingFaceTB/SmolLM2-360M-Instruct")
 print("Loading finished.")