Spaces:
Sleeping
Sleeping
armandstrickernlp
commited on
Commit
·
caeef30
1
Parent(s):
3ef14f8
Make app
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
2 |
|
3 |
-
model_name = 'gpt2-TOD_finetuned_SGD'
|
4 |
tokenizer_TOD = AutoTokenizer.from_pretrained(model_name)
|
5 |
model_TOD = AutoModelForCausalLM.from_pretrained(model_name)
|
6 |
|
|
|
1 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
2 |
|
3 |
+
model_name = 'armandnlp/gpt2-TOD_finetuned_SGD'
|
4 |
tokenizer_TOD = AutoTokenizer.from_pretrained(model_name)
|
5 |
model_TOD = AutoModelForCausalLM.from_pretrained(model_name)
|
6 |
|