Spaces:

Jaehan
/

Text-Generation-1

Sleeping

Jaehan commited on Jun 24, 2023

Commit

b64cf01

1 Parent(s): 9a9d11b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 def generate(text):
     token_ids = tokenizer.encode(text, return_tensors="pt")
-    gpt2_tensors = model.gnerate(token_ids)
     response = gpt2_tensors
     return response

 def generate(text):
     token_ids = tokenizer.encode(text, return_tensors="pt")
+    gpt2_tensors = model.generate(token_ids)
     response = gpt2_tensors
     return response