Spaces:
Build error
Build error
ShAnSantosh
commited on
Commit
•
099c56e
1
Parent(s):
be97f1b
Update app.py
Browse files
app.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
-
from transformers import AutoModelForCausalLM, AutoTokenizer
|
2 |
import torch
|
3 |
-
|
4 |
tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
|
5 |
model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
|
6 |
-
|
7 |
def predict(input, history=[]):
|
8 |
"""
|
9 |
# tokenize the new input sentence
|
|
|
1 |
+
#from transformers import AutoModelForCausalLM, AutoTokenizer
|
2 |
import torch
|
3 |
+
"""
|
4 |
tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
|
5 |
model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
|
6 |
+
"""
|
7 |
def predict(input, history=[]):
|
8 |
"""
|
9 |
# tokenize the new input sentence
|