Update app.py

app.py

@@ -107,7 +107,7 @@ def finbert(word):
         data['Cluster'] = pd.Series(assigned_clusters, index = data.index)
         data['Centroid'] = data['Cluster'].apply(lambda x: Kclusterer.means()[x])
     
-    # return the text if clustering algorithm catches an exceptiona and move to the next text file
+    # return the text if the clustering algorithm catches an exception and move to the next text file
     except ValueError:
         return text
 
@@ -122,7 +122,7 @@ def finbert(word):
     summary = " ".join(data.sort_values(
                 'Distance_From_Centroid',
                 ascending = True).groupby('Cluster').head(1).sort_index()['Sentences'].tolist())
-    return ("FinBERT MODEL OUTPUT:--->"+summary,"    Length of Input:---->"+str(len(word)),"  Length of Output:----> "+str(len(summary)))
+    return (summary,"    Length of Input:---->"+str(len(word)),"  Length of Output:----> "+str(len(summary)))
     
 
 
@@ -196,7 +196,7 @@ def pegasus(text):
     summary = [sentence for each in summary for sentence in each]
     final = "".join(summary)
     
-    return ("FinBERT MODEL OUTPUT:--->"+final,"    Length of Input:---->"+str(len(text)),"  Length of Output:----> "+str(len(final)))
+    return (final,"    Length of Input:---->"+str(len(text)),"  Length of Output:----> "+str(len(final)))
 
 
 import gradio as gr