Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -10,9 +10,9 @@ with st.spinner('Loading Model... (This may take a while)'):
|
|
10 |
st.success('Model loaded correctly!')
|
11 |
|
12 |
prompt= st.text_area('Your prompt here',
|
13 |
-
'''
|
14 |
|
15 |
-
answer = generator(prompt, max_length=
|
16 |
|
17 |
lst = answer[0]['generated_text']
|
18 |
|
|
|
10 |
st.success('Model loaded correctly!')
|
11 |
|
12 |
prompt= st.text_area('Your prompt here',
|
13 |
+
'''AI will help humanity?''')
|
14 |
|
15 |
+
answer = generator(prompt, max_length=60, no_repeat_ngram_size=2, early_stopping=True, num_beams=5)
|
16 |
|
17 |
lst = answer[0]['generated_text']
|
18 |
|