Spaces:
Sleeping
Sleeping
Commit
·
796cc88
1
Parent(s):
ba9c49b
Update app.py
Browse files
app.py
CHANGED
@@ -8,14 +8,13 @@ description = "A State-of-the-Art Large-scale Pretrained Response generation mod
|
|
8 |
examples = [["How are you?"]]
|
9 |
|
10 |
|
11 |
-
|
12 |
-
|
13 |
# tokenizer = LlamaTokenizer.from_pretrained("hf-internal-testing/llama-tokenizer")
|
14 |
# model = LlamaForCausalLM.from_pretrained("hf-internal-testing/llama-tokenizer")
|
15 |
|
16 |
-
model = "meta-llama/Llama-2-7b-chat-hf"
|
17 |
-
|
18 |
-
tokenizer = AutoTokenizer.from_pretrained(model)
|
19 |
|
20 |
|
21 |
def predict(input, history=[]):
|
|
|
8 |
examples = [["How are you?"]]
|
9 |
|
10 |
|
11 |
+
tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
|
12 |
+
model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
|
13 |
# tokenizer = LlamaTokenizer.from_pretrained("hf-internal-testing/llama-tokenizer")
|
14 |
# model = LlamaForCausalLM.from_pretrained("hf-internal-testing/llama-tokenizer")
|
15 |
|
16 |
+
#model = "meta-llama/Llama-2-7b-chat-hf"
|
17 |
+
#tokenizer = AutoTokenizer.from_pretrained(model)
|
|
|
18 |
|
19 |
|
20 |
def predict(input, history=[]):
|