azizmma commited on
Commit
f6a954e
1 Parent(s): 243e741

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,9 +2,10 @@ import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
3
 
4
  model_name = "gpt2-large"
5
- tokenizer = AutoTokenizer.from_pretrained(model_name)
6
  @st.cache
7
  def load_pipeline(model_name):
 
8
  model = AutoModelForCausalLM.from_pretrained(model_name,pad_token_id=tokenizer.eos_token_id)
9
  return pipeline('text-generation', model=model, tokenizer=tokenizer)
10
 
 
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
3
 
4
  model_name = "gpt2-large"
5
+
6
  @st.cache
7
  def load_pipeline(model_name):
8
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
9
  model = AutoModelForCausalLM.from_pretrained(model_name,pad_token_id=tokenizer.eos_token_id)
10
  return pipeline('text-generation', model=model, tokenizer=tokenizer)
11