Eun0 commited on
Commit
f9767c2
β€’
1 Parent(s): 4561634

Remove unused package

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -4,7 +4,6 @@ import pandas as pd
4
  import faiss
5
  import clip
6
  import torch
7
- from datasets import load_dataset
8
 
9
  title = r"""
10
  <h1 align="center" id="space-title"> πŸ” Search Similar Text/Image in the Dataset</h1>
@@ -88,9 +87,10 @@ def search_text(dataset, top_k, show_score, query_text, device):
88
  result_str += f", {score:0.2f}"
89
  result_str += "\n"
90
 
91
- file_name = query_text.replace(" ", "_")
92
- if show_score:
93
- file_name += "_score"
 
94
  output_path = f"./{file_name}.txt"
95
  with open(output_path, "w") as f:
96
  f.writelines(result_str)
 
4
  import faiss
5
  import clip
6
  import torch
 
7
 
8
  title = r"""
9
  <h1 align="center" id="space-title"> πŸ” Search Similar Text/Image in the Dataset</h1>
 
87
  result_str += f", {score:0.2f}"
88
  result_str += "\n"
89
 
90
+ # file_name = query_text.replace(" ", "_")
91
+ # if show_score:
92
+ # file_name += "_score"
93
+ file_name = "output"
94
  output_path = f"./{file_name}.txt"
95
  with open(output_path, "w") as f:
96
  f.writelines(result_str)