micole66 commited on
Commit
48ff222
1 Parent(s): c9f768f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ def get_pipe(name):
4
  tokenizer = AutoTokenizer.from_pretrained(name)
5
  model = AutoModelForSeq2SeqLM.from_pretrained(name)
6
  pipe = pipeline(
7
- "text-generation", model=model, tokenizer=tokenizer, framework="pt"
8
  )
9
  return pipe
10
  model_names = ['EleutherAI/gpt-neo-1.3B'] #, 'bigscience/T0p', 'bigscience/T0pp']
 
4
  tokenizer = AutoTokenizer.from_pretrained(name)
5
  model = AutoModelForSeq2SeqLM.from_pretrained(name)
6
  pipe = pipeline(
7
+ "text-generator", model=model, tokenizer=tokenizer, framework="pt"
8
  )
9
  return pipe
10
  model_names = ['EleutherAI/gpt-neo-1.3B'] #, 'bigscience/T0p', 'bigscience/T0pp']