Spaces:

m-ric
/

beam_search_visualizer

Running

m-ric commited on Mar 25, 2024

Commit

89b8c0b

verified ·

1 Parent(s): 85a45ca

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("mistralai/Mistral-7B-Instruct-v0.2")
-model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
 print("Loading finished.")

 import gradio as gr
 import spaces
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
+model = AutoModelForCausalLM.from_pretrained("gpt2")
 print("Loading finished.")