Spaces:

micole66
/

test

Runtime error

micole66 commited on Feb 10, 2022

Commit

48ff222

1 Parent(s): c9f768f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ def get_pipe(name):
     tokenizer = AutoTokenizer.from_pretrained(name)
     model = AutoModelForSeq2SeqLM.from_pretrained(name)
     pipe = pipeline(
-	"text-generation", model=model, tokenizer=tokenizer, framework="pt"
     )
     return pipe
 model_names = ['EleutherAI/gpt-neo-1.3B'] #, 'bigscience/T0p', 'bigscience/T0pp']

     tokenizer = AutoTokenizer.from_pretrained(name)
     model = AutoModelForSeq2SeqLM.from_pretrained(name)
     pipe = pipeline(
+	"text-generator", model=model, tokenizer=tokenizer, framework="pt"
     )
     return pipe
 model_names = ['EleutherAI/gpt-neo-1.3B'] #, 'bigscience/T0p', 'bigscience/T0pp']