Geolumina
/

instructor-xl

Model card Files Files and versions Community

kblevins commited on Mar 5

Commit

6c69c8a

•

1 Parent(s): e11605b

updated handler

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -25,4 +25,4 @@ class EndpointHandler:
         # encode
         embedding = self.model.encode(query)
-        return embedding[0]

         # encode
         embedding = self.model.encode(query)
+        return embedding