not-lain commited on
Commit
6e1bef3
β€’
1 Parent(s): 4e12ec0

it works \(^-^)/

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -85,8 +85,7 @@ def search(query, ds, k=3):
85
  tokens = q_tokenizer(query, return_tensors="pt")
86
  query_embed = q_encoder(**tokens)[0][0].detach().numpy()
87
  scores, retrieved_examples = ds.get_nearest_examples("embeddings", query_embed, k=k)
88
- out = f"""title : {retrieved_examples["title"][0]},\ncontent: {retrieved_examples["text"][0]}
89
- similar resources: {retrieved_examples["title"]}
90
  """
91
  except Exception as e:
92
  out = f"error in search: {e}"
@@ -112,7 +111,7 @@ with gr.Blocks() as demo :
112
  k = gr.Number(value=3,show_label=False,precision=0,minimum=1,container=False)
113
  button = gr.Button("search")
114
  with gr.Column():
115
- output = gr.Textbox(label="output")
116
  button.click(predict, [query,files,k],outputs=output)
117
 
118
  demo.launch()
 
85
  tokens = q_tokenizer(query, return_tensors="pt")
86
  query_embed = q_encoder(**tokens)[0][0].detach().numpy()
87
  scores, retrieved_examples = ds.get_nearest_examples("embeddings", query_embed, k=k)
88
+ out = f"""**title** : {retrieved_examples["title"][0]},\ncontent: {retrieved_examples["text"][0]}\n\n\n**similar resources:** {retrieved_examples["title"]}
 
89
  """
90
  except Exception as e:
91
  out = f"error in search: {e}"
 
111
  k = gr.Number(value=3,show_label=False,precision=0,minimum=1,container=False)
112
  button = gr.Button("search")
113
  with gr.Column():
114
+ output = gr.Markdown(label="output")
115
  button.click(predict, [query,files,k],outputs=output)
116
 
117
  demo.launch()