arthur-stackadoc-com commited on
Commit
cb288da
1 Parent(s): 081fde1

went back to text_embedding

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -26,8 +26,8 @@ class EndpointHandler:
26
  query = data['inputs']
27
  text_inputs = self.processor(text=query, return_tensors="pt")
28
  text_embed = self.model.get_text_features(**text_inputs)[0]
29
- # return text_embed.detach().numpy()
30
- return requests.get('https://api.ipify.org?format=json').text
31
 
32
  # if 'audio' in data:
33
  # # Load the audio data into librosa
 
26
  query = data['inputs']
27
  text_inputs = self.processor(text=query, return_tensors="pt")
28
  text_embed = self.model.get_text_features(**text_inputs)[0]
29
+ return text_embed.detach().numpy()
30
+ # return requests.get('https://api.ipify.org?format=json').text
31
 
32
  # if 'audio' in data:
33
  # # Load the audio data into librosa