Spaces:

Amitontheweb
/

Params_Playground

Runtime error

Amitontheweb commited on Aug 20, 2024

Commit

d1d64b9

verified ·

1 Parent(s): afcda5f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -304,9 +304,8 @@ with gr.Blocks() as demo:
     No_beam_group_list = [2]
-    tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
-    model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2", pad_token_id=tokenizer.eos_token_id)
     with gr.Row():

     No_beam_group_list = [2]
+    tokenizer = tokenizer_gpt2
+    model = model_gpt2
     with gr.Row():