Spaces:

Jaehan
/

Text-Generation-1

Sleeping

Jaehan commited on Jun 24, 2023

Commit

cfcd64c

1 Parent(s): 8049e5d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import gradio as gr
 model_name = "gpt2"
-model = GPT2LMHeaddModel.from_pretrained(model_name)
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 def generate(text):

 import gradio as gr
 model_name = "gpt2"
+model = GPT2LMHeadModel.from_pretrained(model_name)
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 def generate(text):