Update app.py

app.py

@@ -169,26 +169,32 @@ def create_embeddings(texts_to_embedd):
 
 
 def predict(text_input, progress=gr.Progress()):
-    
+
+    progress(0.1, desc="Getting queried data...")
     # get data.
     records_df = query_records(text_input)
     print(records_df)
     
     
-    
+    progress(0.2, desc="Embedding Data...")
     texts_to_embedd = [title + tokenizer.sep_token + publication + tokenizer.sep_token  + abstract for title, publication, abstract in zip(records_df['title'],records_df['parsed_publication'], records_df['abstract'])]
     
     embeddings = create_embeddings(texts_to_embedd)
     print(embeddings)
+
+    progress(0.3, desc="Project into UMAP-embedding...")
     umap_embeddings = mapper.transform(embeddings)
     records_df[['x','y']] = umap_embeddings
 
     basedata_df['color'] = '#ced4d211'
     records_df['color'] = '#a81a26'
 
+    progress(0.4, desc="Set up data...")
+
     stacked_df = pd.concat([records_df, basedata_df], axis=0, ignore_index=True)
     stacked_df = stacked_df.fillna("Unlabelled")
-
+    stacked_df = stacked_df.reset_index(drop=True)
+    print(stacked_df)
 
     
     
@@ -199,7 +205,7 @@ def predict(text_input, progress=gr.Progress()):
  
    #
     
-    progress(0.7, desc="Loading hover data...")
+    progress(0.5, desc="Plotting...")
 
     plot = datamapplot.create_interactive_plot(
         stacked_df[['x','y']].values,
@@ -208,7 +214,7 @@ def predict(text_input, progress=gr.Progress()):
         font_family="Roboto Condensed",marker_color_array=stacked_df['color']
     )
 
-    progress(0.9, desc="Saving plot...")
+    progress(0.6, desc="Saving plot...")
     plot.save(file_path)
 
     progress(1.0, desc="Done!")