Update app.py

app.py

@@ -19,12 +19,13 @@ model_name = 'nlpaueb/legal-bert-base-uncased'
 #model_name = 'laxya007/gpt2_legal'
 # model_name = 'facebook/bart-large-cnn'
 
-# The setup of huggingface.co
-custom_config = AutoConfig.from_pretrained(model_name)
-custom_config.output_hidden_states=True
-custom_tokenizer = AutoTokenizer.from_pretrained(model_name)
-custom_model = AutoModel.from_pretrained(model_name, config=custom_config)
-bert_legal_model = Summarizer(custom_model=custom_model, custom_tokenizer=custom_tokenizer)
+
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("laxya007/gpt2_BSA_Legal_Initiproject_OE_OS_BRM")
+
+model = AutoModelForCausalLM.from_pretrained("laxya007/gpt2_BSA_Legal_Initiproject_OE_OS_BRM")
+bert_legal_model = Summarizer(custom_model= model, custom_tokenizer= tokenizer)
 print('Using model {}\n'.format(model_name))