Spaces:

nihaldsouza1
/

clearlydefined_license_summarizer

Runtime error

Nihal D'Souza commited on May 21, 2022

Commit

0599777

1 Parent(s): cedd239

Removing loading message banner

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ nltk.download('punkt')
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
-with st.spinner('Loading model from Huggingface...'):
     model = AutoModelForSeq2SeqLM.from_pretrained(CUSTOM_MODEL_NAME).to(device)
     tokenizer = AutoTokenizer.from_pretrained(CUSTOM_MODEL_NAME)

 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+with st.spinner('Loading...'):
     model = AutoModelForSeq2SeqLM.from_pretrained(CUSTOM_MODEL_NAME).to(device)
     tokenizer = AutoTokenizer.from_pretrained(CUSTOM_MODEL_NAME)