ArthurSynthia
/

clap1

Feature Extraction

Inference Endpoints

Model card Files Files and versions Community

arthur-stackadoc-com commited on May 15, 2024

Commit

206504f

·

1 Parent(s): 202b18b

handler.py : JSON SERIALZATION

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -21,4 +21,4 @@ class EndpointHandler:
         text_inputs = self.processor(text=query, return_tensors="pt")
         text_embed = self.model.get_text_features(**text_inputs)[0]
         gc.collect()
-        return text_embed

         text_inputs = self.processor(text=query, return_tensors="pt")
         text_embed = self.model.get_text_features(**text_inputs)[0]
         gc.collect()
+        return text_embed.detach().numpy()