Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ from fudge.constants import *
|
|
29 |
|
30 |
|
31 |
# imp.reload(model)
|
32 |
-
pretrained_model = "
|
33 |
generation_model = AutoModelForSeq2SeqLM.from_pretrained(pretrained_model, return_dict=True).to(device)
|
34 |
|
35 |
device = 'cuda'
|
@@ -40,7 +40,7 @@ generation_model.eval()
|
|
40 |
model_args = mock.Mock()
|
41 |
model_args.task = 'clickbait'
|
42 |
model_args.device = device
|
43 |
-
model_args.checkpoint = '
|
44 |
|
45 |
# conditioning_model = Model(model_args, pad_id, len(dataset_info.index2word)) # no need to get the glove embeddings when reloading since they're saved in model ckpt anyway
|
46 |
conditioning_model = Model(model_args, pad_id, vocab_size=None) # no need to get the glove embeddings when reloading since they're saved in model ckpt anyway
|
|
|
29 |
|
30 |
|
31 |
# imp.reload(model)
|
32 |
+
pretrained_model = "checkpoint-150/"
|
33 |
generation_model = AutoModelForSeq2SeqLM.from_pretrained(pretrained_model, return_dict=True).to(device)
|
34 |
|
35 |
device = 'cuda'
|
|
|
40 |
model_args = mock.Mock()
|
41 |
model_args.task = 'clickbait'
|
42 |
model_args.device = device
|
43 |
+
model_args.checkpoint = 'checkpoint-1464/'
|
44 |
|
45 |
# conditioning_model = Model(model_args, pad_id, len(dataset_info.index2word)) # no need to get the glove embeddings when reloading since they're saved in model ckpt anyway
|
46 |
conditioning_model = Model(model_args, pad_id, vocab_size=None) # no need to get the glove embeddings when reloading since they're saved in model ckpt anyway
|