Update app.py

app.py

@@ -99,10 +99,9 @@ def compare_text(transcript, categories, threshold):
     fig.update_traces(textfont_size=24, textangle=0, textposition="inside", cliponaxis=False)
     fig.update_yaxes(range=[0, 1])  
 
-    # details = df_results #.drop(labels='line',axis=1).sort_values(['tag','similarity'],ascending=[True,False]).groupby('tag').head(3).reset_index()    .drop(labels='index',axis=1)
-  
-    res = df_summary['similarity'].round(decimals = 2).to_dict()    
-    return res, fig, df_results.round(decimals = 3)
+    df_results = df_results.round(decimals = 3) # .drop(labels='line',axis=1).groupby('tag').head(3).reset_index()    .drop(labels='index',axis=1)
+    res = df_summary['similarity'].round(decimals = 2).to_dict()
+    return res, fig, df_results
 
 categories = """Hello=Hello, how are you doing today?;Hi, everybody;Hi;My name's Johnny
 What=most advanced conversation intelligence and AI powered coaching platform;a software platform that helps people reach their potential;for communicating and connecting;empowered by behavioral science;uses artificial intelligence;drives performance outcomes for customer facing teams;help them sell more;help them deliver better experiences
@@ -114,7 +113,7 @@ Bye=Bye;And I look forward to talking to you again soon. Thank you very much"""
 threshold_default = 0.50
 
 with gr.Blocks(css=".gradio-container { background-color: white; background-image: url('file=./qc-logo.png'); background-size: 75px 75px; background-repeat: no-repeat; background-position: 0px 0px; }") as demo:
-    gr.Markdown(f"# {' ' * 8}Transcript classifier with Roberta")
+    gr.Markdown(f"# {' ' * 8}Transcript to BERT paraphrase-distilroberta-base-v2")
     with gr.Row():
         source = gr.Textbox(lines=3, label="Transcript", placeholder="Source text")
     with gr.Row():