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

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")
18
  inputs = tokenizer.encode(input_sequence, return_tensors='pt')
19
 
20
  # generate text until the output length (which includes the context length) reaches 50