Update app.py

app.py

@@ -15,17 +15,17 @@ from summarizer import Summarizer,TransformerSummarizer
 from transformers import pipelines
 nltk.download('punkt')
 
+
+model_name = 'nlpaueb/legal-bert-base-uncased'
+
+
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+model = AutoModelForPreTraining.from_pretrained(model_name)
+bert_legal_model = Summarizer(custom_model=model, custom_tokenizer=tokenizer)
  
 
 def get_response(input_text):
-  model_name = 'nlpaueb/legal-bert-base-uncased'
-  
-  
-  tokenizer = AutoTokenizer.from_pretrained(model_name )
-  
-  model = AutoModelForPreTraining.from_pretrained(model_name)
-  bert_legal_model = Summarizer(custom_model=model, custom_tokenizer=tokenizer)
-  
 
   output_text= bert_legal_model(input_text,  min_length = 8, ratio = 0.05)
   return output_text
@@ -39,4 +39,4 @@ iface = gr.Interface(
    )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(share=True)