WolfeLeo2 commited on
Commit
9d694c5
Β·
1 Parent(s): 0cc7bc2

first commit

Browse files
Files changed (2) hide show
  1. app.py +42 -0
  2. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import logging
3
+ import sys
4
+ from transformers import pipeline
5
+
6
+ # Configure logging
7
+ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
8
+ logger = logging.getLogger(__name__)
9
+
10
+ # Load the model
11
+ logger.info("Loading bart-large-cnn model...")
12
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
13
+ logger.info("Model loaded successfully!")
14
+
15
+ def summarize_text(text, max_length=150, min_length=30):
16
+ if not text or len(text.strip()) < 50:
17
+ return text
18
+
19
+ logger.info(f"Summarizing text of length {len(text)}")
20
+ result = summarizer(
21
+ text,
22
+ max_length=max_length,
23
+ min_length=min_length,
24
+ truncation=True
25
+ )
26
+ summary = result[0]["summary_text"]
27
+ return summary
28
+
29
+ # Create Gradio interface
30
+ demo = gr.Interface(
31
+ fn=summarize_text,
32
+ inputs=[
33
+ gr.Textbox(lines=10, label="Text to Summarize"),
34
+ gr.Slider(50, 500, value=150, label="Max Length"),
35
+ gr.Slider(10, 200, value=30, label="Min Length")
36
+ ],
37
+ outputs=gr.Textbox(label="Summary"),
38
+ title="StudAI ",
39
+ description="Powered by facebook/bart-large-cnn model"
40
+ )
41
+
42
+ demo.launch(share=True, server_name="Leo", server_port=7860)
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ gradio==4.13.0
2
+ transformers==4.35.2
3
+ torch==2.0.1