0xalfroz commited on
Commit
d1f2845
·
verified ·
1 Parent(s): 0c07570

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -7,10 +7,15 @@ model = AutoModel.from_pretrained(model_name)
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
 
9
  def text_to_vector(text):
 
 
 
10
  inputs = tokenizer(text, return_tensors="pt")
11
  outputs = model(**inputs)
12
  vector = outputs.pooler_output.detach().numpy()[0]
13
- return vector
 
 
14
 
15
  demo = gr.Interface(
16
  fn=text_to_vector,
 
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
 
9
  def text_to_vector(text):
10
+ """
11
+ Converts text to a vector representation using a pre-trained model.
12
+ """
13
  inputs = tokenizer(text, return_tensors="pt")
14
  outputs = model(**inputs)
15
  vector = outputs.pooler_output.detach().numpy()[0]
16
+ # Convert the vector to a string representation for display
17
+ vector_str = ", ".join([str(x) for x in vector])
18
+ return vector_str
19
 
20
  demo = gr.Interface(
21
  fn=text_to_vector,