max21226 commited on
Commit
ad0d66d
1 Parent(s): 7be2597
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import streamlit as st
2
  from transformers import pipeline, set_seed
3
- from datetime import date
4
 
5
  def init():
6
  generator = setupAI()
@@ -8,7 +8,7 @@ def init():
8
 
9
  def setupAI():
10
  generator = pipeline('text-generation', model='gpt2')
11
- set_seed(date.datetime.now().Second)
12
  return generator
13
 
14
  generator = init()
 
1
  import streamlit as st
2
  from transformers import pipeline, set_seed
3
+ from datetime import datetime
4
 
5
  def init():
6
  generator = setupAI()
 
8
 
9
  def setupAI():
10
  generator = pipeline('text-generation', model='gpt2')
11
+ set_seed(datetime.now().second)
12
  return generator
13
 
14
  generator = init()