Spaces:

Funbi
/

Textgen

Runtime error

Funbi commited on Oct 11, 2022

Commit

c9f4ade

1 Parent(s): 3fc6414

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ tokenizer = GPT2Tokenizer.from_pretrained("gpt2-large")
 model = GPT2LMHeadModel.from_pretrained("gpt2-large", pad_token_id=tokenizer.eos_token_id)
-def generate(prompt,textCount):
     input_ids = tokenizer.encode(prompt, return_tensors='pt')
     if textCount == None or textCount < 200:
         textCount=200

 model = GPT2LMHeadModel.from_pretrained("gpt2-large", pad_token_id=tokenizer.eos_token_id)
+def generate(prompt,textCount=200):
     input_ids = tokenizer.encode(prompt, return_tensors='pt')
     if textCount == None or textCount < 200:
         textCount=200