Update app.py

app.py

@@ -345,9 +345,9 @@ def load_or_finetune_legalbert():
     return tokenizer, model
 
 # Load models
-pegasus_tokenizer, pegasus_model = load_or_finetune_pegasus()
-bert_tokenizer, bert_model = load_or_finetune_bert()
-legalbert_tokenizer, legalbert_model = load_or_finetune_legalbert()
+# pegasus_tokenizer, pegasus_model = load_or_finetune_pegasus()
+# bert_tokenizer, bert_model = load_or_finetune_bert()
+# legalbert_tokenizer, legalbert_model = load_or_finetune_legalbert()
 
 def extract_text_from_pdf(file_path):
     text = ""
@@ -405,6 +405,11 @@ def summarize_with_legalbert(text):
     key_sentence_idx = probs.argsort(descending=True)[:5]
     return ". ".join([sentences[idx] for idx in key_sentence_idx if sentences[idx].strip()])
 
+# Load Models
+pegasus_tokenizer, pegasus_model = load_or_finetune_pegasus()
+bert_tokenizer, bert_model = load_or_finetune_bert()
+legalbert_tokenizer, legalbert_model = load_or_finetune_legalbert()
+
 @app.route('/summarize', methods=['POST'])
 def summarize_document():
     if 'file' not in request.files:
@@ -455,4 +460,6 @@ def summarize_document():
     return jsonify({"model_used": model, "summary": summary})
 
 if __name__ == '__main__':
-    app.run(debug=True, host='0.0.0.0', port=7860)
+    # For local development, use Flask's built-in server
+    port = int(os.environ.get("PORT", 5000))  # Use PORT env var if set (common in containers)
+    app.run(debug=False, host='0.0.0.0', port=port)  # debug=False for production-like behavior