Update app.py

app.py

@@ -85,33 +85,24 @@ def calculate_column_widths(df):
     return column_widths
 
 
-def show_results_tab(df):
+def show_results_tab(df, model_info, results):
     with gr.Row():
         model_name = gr.Textbox(value='Input the Model Name (fuzzy)', label='Model Name')
 
 
     def filter_df(model_name):
-
-        df = generate_table(results)
-
-        df['flag'] = [model_size_flag(x, model_size) for x in df['Param (B)']]
-        df = df[df['flag']]
-        df.pop('flag')
-        if len(df):
-            df['flag'] = [model_type_flag(df.iloc[i], model_type) for i in range(len(df))]
-            df = df[df['flag']]
-            df.pop('flag')
-        df['Rank'] = list(range(1, len(df) + 1))
+        df = make_results_tab(model_info, results)
         default_val = 'Input the Model Name (fuzzy)'
         if model_name != default_val:
-            method_names = [x.split('</a>')[0].split('>')[-1].lower() for x in df['Method']]
+            method_names = [x.split('</a>')[0].split('>')[-1].lower() for x in df['Model Name']]
             flag = [model_name.lower() in name for name in method_names]
             df['TEMP'] = flag
             df = df[df['TEMP'] == True] 
             df.pop('TEMP')
+    return df
+
+    model_name.submit(fn=filter_df, inputs=[model_name,], outputs=df)
 
-        comp = gr.components.DataFrame(value=df[headers], datatype=[type_map[x] for x in headers])
-    return comp
 
     with gr.Column():
         table = gr.DataFrame(
@@ -122,6 +113,7 @@ def show_results_tab(df):
     )
 
 
+
 def create_interface():
     model_info, results = findfile()
 
@@ -131,7 +123,7 @@ def create_interface():
         with gr.Tabs(elem_classes='tab-buttons') as tabs:
             with gr.TabItem('Results', elem_id='main', id=0):
                 df = make_results_tab(model_info, results)
-                show_results_tab(df)
+                show_results_tab(df, model_info, results)
 
             with gr.TabItem('Predictions', elem_id='notmain', id=1):
                 # dataset_tab(results, structs[i], dataset)
@@ -146,4 +138,3 @@ if __name__ == '__main__':
     demo = create_interface()
     demo.queue()
     demo.launch(server_name='0.0.0.0')
-