Spaces:
Running
Running
kingabzpro
commited on
Commit
•
f72e3da
1
Parent(s):
1b2f46a
Update app.py
Browse files
app.py
CHANGED
@@ -18,8 +18,8 @@ article = "<p style='text-align: center'><a href='https://medium.com/geekculture
|
|
18 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
19 |
import torch
|
20 |
|
21 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
22 |
-
model = AutoModelForCausalLM.from_pretrained("
|
23 |
|
24 |
def predict(input, history=[]):
|
25 |
# tokenize the new input sentence
|
|
|
18 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
19 |
import torch
|
20 |
|
21 |
+
tokenizer = AutoTokenizer.from_pretrained("ericzhou/DialoGPT-Medium-Rick_v2")
|
22 |
+
model = AutoModelForCausalLM.from_pretrained("ericzhou/DialoGPT-Medium-Rick_v2")
|
23 |
|
24 |
def predict(input, history=[]):
|
25 |
# tokenize the new input sentence
|