yash161101 commited on
Commit
7dcf1a6
1 Parent(s): d68e3e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ MAX_LEN = number
14
  if st.button('Submit'):
15
 
16
  tokenizer = AutoTokenizer.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
17
- model = AutoModelForCausalLM.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator",args={'fp16': False}))
18
  inputs = tokenizer.encode(input_sequence, return_tensors='pt')
19
 
20
  # generate text until the output length (which includes the context length) reaches 50
 
14
  if st.button('Submit'):
15
 
16
  tokenizer = AutoTokenizer.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
17
+ model = AutoModelForCausalLM.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator",args={'fp16': False})
18
  inputs = tokenizer.encode(input_sequence, return_tensors='pt')
19
 
20
  # generate text until the output length (which includes the context length) reaches 50