Commit ·
206504f
1
Parent(s): 202b18b
handler.py : JSON SERIALZATION
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -21,4 +21,4 @@ class EndpointHandler:
|
|
| 21 |
text_inputs = self.processor(text=query, return_tensors="pt")
|
| 22 |
text_embed = self.model.get_text_features(**text_inputs)[0]
|
| 23 |
gc.collect()
|
| 24 |
-
return text_embed
|
|
|
|
| 21 |
text_inputs = self.processor(text=query, return_tensors="pt")
|
| 22 |
text_embed = self.model.get_text_features(**text_inputs)[0]
|
| 23 |
gc.collect()
|
| 24 |
+
return text_embed.detach().numpy()
|