Spaces:
Sleeping
Sleeping
Commit
·
a5d1018
1
Parent(s):
10764a4
Update app.py
Browse files
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:
|