Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
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
|