Spaces:
Runtime error
Runtime error
yash161101
commited on
Commit
•
d68e3e5
1
Parent(s):
ee02dc5
Update app.py
Browse files
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",
|
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
|