deniandriancode commited on
Commit
a5d1018
·
1 Parent(s): 10764a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
- from transformers import pipeline, Conversation, BlenderbotTokenizer, BlenderbotForConditionalGeneration
 
3
 
4
  tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
5
  model = BlenderbotForConditionalGeneration.from_pretrained("model")
@@ -9,6 +10,8 @@ chatbot = pipeline("conversational", model=model, tokenizer=tokenizer)
9
  converse = None
10
 
11
  def model_generate(text, state = None):
 
 
12
  if state == None:
13
  converse = Conversation(text)
14
  else:
 
1
  import gradio as gr
2
+ from transformers import pipeline, Conversation, BlenderbotTokenizer, BlenderbotForConditionalGeneration, set_seed
3
+ import random
4
 
5
  tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
6
  model = BlenderbotForConditionalGeneration.from_pretrained("model")
 
10
  converse = None
11
 
12
  def model_generate(text, state = None):
13
+ seed = random.randint(100)
14
+ set_seed(seed)
15
  if state == None:
16
  converse = Conversation(text)
17
  else: