Spaces:

csuvikv
/

gpt-j-6b

Runtime error

csuvikv commited on Aug 30, 2023

Commit

5c1e5a2

1 Parent(s): 01bebe8

function arg

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,8 +8,8 @@ model = GPTNeoForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
 tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
-def Bemenet(bemenet):
-    input_ids = tokenizer(bemenet, return_tensors="pt").input_ids
     gen_tokens = model.generate(
         input_ids,

 tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
+def Bemenet(input_string):
+    input_ids = tokenizer(input_string, return_tensors="pt").input_ids
     gen_tokens = model.generate(
         input_ids,