Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,7 @@ from summarization_methods import summary_with_text_reduction
|
|
4 |
from summarization_methods import summary_with_tfidf
|
5 |
import gradio as gr
|
6 |
from enum import Enum
|
|
|
7 |
|
8 |
class SummarizationMehods(Enum):
|
9 |
LSA = 0
|
@@ -13,6 +14,7 @@ class SummarizationMehods(Enum):
|
|
13 |
|
14 |
|
15 |
def summary(text , num_sentences=3 , method = SummarizationMehods(0).name):
|
|
|
16 |
if method.casefold() == SummarizationMehods(0).name.casefold():
|
17 |
summary = summary_with_lsa(text , num_sentences)
|
18 |
elif method.casefold()==SummarizationMehods(1).name.casefold():
|
|
|
4 |
from summarization_methods import summary_with_tfidf
|
5 |
import gradio as gr
|
6 |
from enum import Enum
|
7 |
+
from preprocess import clean_text
|
8 |
|
9 |
class SummarizationMehods(Enum):
|
10 |
LSA = 0
|
|
|
14 |
|
15 |
|
16 |
def summary(text , num_sentences=3 , method = SummarizationMehods(0).name):
|
17 |
+
text = clean_text(text)
|
18 |
if method.casefold() == SummarizationMehods(0).name.casefold():
|
19 |
summary = summary_with_lsa(text , num_sentences)
|
20 |
elif method.casefold()==SummarizationMehods(1).name.casefold():
|