deniandriancode commited on
Commit
8cb5991
·
1 Parent(s): 2f1f99d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,5 +1,5 @@
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")
@@ -10,8 +10,6 @@ chatbot = pipeline("conversational", model=model, tokenizer=tokenizer)
10
  converse = None
11
 
12
  def model_generate(text, state = None):
13
- seed = random.randint(0, 101)
14
- set_seed(seed)
15
  if state == None:
16
  converse = Conversation(text)
17
  else:
@@ -21,6 +19,7 @@ def model_generate(text, state = None):
21
  return converse.generated_responses[-1]
22
 
23
  def get_response(text, history) -> str:
 
24
  output = model_generate(text, converse)
25
  return output
26
 
 
1
  import gradio as gr
2
+ from transformers import pipeline, Conversation, BlenderbotTokenizer, BlenderbotForConditionalGeneration
3
  import random
4
 
5
  tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
 
10
  converse = None
11
 
12
  def model_generate(text, state = None):
 
 
13
  if state == None:
14
  converse = Conversation(text)
15
  else:
 
19
  return converse.generated_responses[-1]
20
 
21
  def get_response(text, history) -> str:
22
+ chatbot.generator.set_seed(random.randint(0, 101))
23
  output = model_generate(text, converse)
24
  return output
25