Geolumina
/

instructor-xl

Model card Files Files and versions Community

kblevins commited on Mar 5

Commit

e11605b

•

1 Parent(s): 3bdedb5

print inputs

Files changed (1) hide show

handler.py +3 -1

handler.py CHANGED Viewed

@@ -16,9 +16,11 @@ class EndpointHandler:
         # get inputs
         inputs: dict = data.pop("inputs", data)
         # add instruction
         query  = [['Retrieve documents that can help answer the question:',
-                   inputs[0]]]
         # encode
         embedding = self.model.encode(query)

         # get inputs
         inputs: dict = data.pop("inputs", data)
+        print(inputs)
         # add instruction
         query  = [['Retrieve documents that can help answer the question:',
+                   inputs]]
         # encode
         embedding = self.model.encode(query)