yash161101 commited on
Commit
8c60512
1 Parent(s): 724a7d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
 
13
  @st.cache(allow_output_mutation=True)
14
  def get_model():
15
- tokenizer = AutoTokenizer.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
16
  model = AutoModelForCausalLM.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
17
  return model, tokenizer
18
 
 
12
 
13
  @st.cache(allow_output_mutation=True)
14
  def get_model():
15
+ tokenizer = AutoTokenizer.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
16
  model = AutoModelForCausalLM.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
17
  return model, tokenizer
18