maahi2412 commited on
Commit
dd36399
·
verified ·
1 Parent(s): da52d57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -345,9 +345,9 @@ def load_or_finetune_legalbert():
345
  return tokenizer, model
346
 
347
  # Load models
348
- pegasus_tokenizer, pegasus_model = load_or_finetune_pegasus()
349
- bert_tokenizer, bert_model = load_or_finetune_bert()
350
- legalbert_tokenizer, legalbert_model = load_or_finetune_legalbert()
351
 
352
  def extract_text_from_pdf(file_path):
353
  text = ""
@@ -405,6 +405,11 @@ def summarize_with_legalbert(text):
405
  key_sentence_idx = probs.argsort(descending=True)[:5]
406
  return ". ".join([sentences[idx] for idx in key_sentence_idx if sentences[idx].strip()])
407
 
 
 
 
 
 
408
  @app.route('/summarize', methods=['POST'])
409
  def summarize_document():
410
  if 'file' not in request.files:
@@ -455,4 +460,6 @@ def summarize_document():
455
  return jsonify({"model_used": model, "summary": summary})
456
 
457
  if __name__ == '__main__':
458
- app.run(debug=True, host='0.0.0.0', port=7860)
 
 
 
345
  return tokenizer, model
346
 
347
  # Load models
348
+ # pegasus_tokenizer, pegasus_model = load_or_finetune_pegasus()
349
+ # bert_tokenizer, bert_model = load_or_finetune_bert()
350
+ # legalbert_tokenizer, legalbert_model = load_or_finetune_legalbert()
351
 
352
  def extract_text_from_pdf(file_path):
353
  text = ""
 
405
  key_sentence_idx = probs.argsort(descending=True)[:5]
406
  return ". ".join([sentences[idx] for idx in key_sentence_idx if sentences[idx].strip()])
407
 
408
+ # Load Models
409
+ pegasus_tokenizer, pegasus_model = load_or_finetune_pegasus()
410
+ bert_tokenizer, bert_model = load_or_finetune_bert()
411
+ legalbert_tokenizer, legalbert_model = load_or_finetune_legalbert()
412
+
413
  @app.route('/summarize', methods=['POST'])
414
  def summarize_document():
415
  if 'file' not in request.files:
 
460
  return jsonify({"model_used": model, "summary": summary})
461
 
462
  if __name__ == '__main__':
463
+ # For local development, use Flask's built-in server
464
+ port = int(os.environ.get("PORT", 5000)) # Use PORT env var if set (common in containers)
465
+ app.run(debug=False, host='0.0.0.0', port=port) # debug=False for production-like behavior