Spaces:

Sanj004
/

Bart-base_Summarizer

Build error

Sanj004 commited on Apr 15, 2024

Commit

bfcb0af

verified ·

1 Parent(s): 0723d6e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import torch
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-SAVED_MODEL_PATH = './bart_base_full_finetune_save'
 model_name = "facebook/bart-base"
 model = AutoModelForSeq2SeqLM.from_pretrained(SAVED_MODEL_PATH).to(device)
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+SAVED_MODEL_PATH = 'bart_base_full_finetune_save'
 model_name = "facebook/bart-base"
 model = AutoModelForSeq2SeqLM.from_pretrained(SAVED_MODEL_PATH).to(device)
 tokenizer = AutoTokenizer.from_pretrained(model_name)