Change index by dataset name only without user namespace

app.py

@@ -88,7 +88,7 @@ def dataset_with_info(dataset):
                     "downloads": dataset.downloads,
                     "created_at": dataset.created_at,
                     "tags": dataset.tags,
-                    "text": f"{dataset.id}-{','.join(list(columns.keys()))}",
+                    "text": f"{str(dataset.id).split('/')[-1]}-{','.join(list(columns.keys()))}",
                 }
     except Exception as e:
         logger.error(f"Failed to get info for {dataset.id}: {e}")
@@ -134,7 +134,6 @@ def search(dataset_name, k):
     print(f"start search for {dataset_name}")
     try:
         dataset_row = all_datasets_df[all_datasets_df.dataset == dataset_name].iloc[0]
-        print(dataset_row)
     except IndexError:
         return pd.DataFrame([{"error": "❌ Dataset does not exist or is not supported"}])
     text = dataset_row["text"]
@@ -143,12 +142,9 @@ def search(dataset_name, k):
     faiss.normalize_L2(_vector)
     distances, ann = index.search(_vector, k=k)
     results = pd.DataFrame({"distances": distances[0], "ann": ann[0]})
-    print("results for distances and ann")
-    print(results)
     merge = pd.merge(results, all_datasets_df, left_on="ann", right_index=True)
-    print("resultst for merged df (distances,ann, dataset info)")
     merge["dataset"] = merge["dataset"].apply(render_model_hub_link)
-    return merge
+    return merge.drop("text", axis=1)
 
 
 with gr.Blocks() as demo: