RickMartel commited on
Commit
3542db6
1 Parent(s): 5d47964

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ class StoppingCriteriaSub(StoppingCriteria):
30
  stop_words = ['.']
31
  stop_words_ids = [tokenizer(stop_word, return_tensors='pt', add_special_tokens=False)['input_ids'].squeeze() for stop_word in stop_words]
32
  stopping_criteria = StoppingCriteriaList([StoppingCriteriaSub(stops=stop_words_ids,
33
- encounters=2)])
34
  st.set_page_config(page_title="GPT2 4 Bible")
35
  st.sidebar.title("GPT2 4 Bible")
36
  st.sidebar.markdown(
 
30
  stop_words = ['.']
31
  stop_words_ids = [tokenizer(stop_word, return_tensors='pt', add_special_tokens=False)['input_ids'].squeeze() for stop_word in stop_words]
32
  stopping_criteria = StoppingCriteriaList([StoppingCriteriaSub(stops=stop_words_ids,
33
+ encounters=3)])
34
  st.set_page_config(page_title="GPT2 4 Bible")
35
  st.sidebar.title("GPT2 4 Bible")
36
  st.sidebar.markdown(