Spaces:
Sleeping
Sleeping
Commit
·
8cb5991
1
Parent(s):
2f1f99d
Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
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,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 |
|