dariadaria
/

reviews_classifier

Text Classification

text-embeddings-inference

Model card Files Files and versions

dariadaria commited on Aug 17, 2023

Commit

3d192a7

·

1 Parent(s): 3702b9d

Update handler.py

Files changed (1) hide show

handler.py +1 -2

handler.py CHANGED Viewed

@@ -37,13 +37,12 @@ class EndpointHandler:
         id = data.pop("id", data)
         text = data.pop("text", data)
         batch = {
-            'id': [],
             'text': [],
             'topic': []
         }
         for topic in topics:
-          batch['id'].append(id)
           batch['text'].append(text)
           batch['topic'].append(topic)

         id = data.pop("id", data)
         text = data.pop("text", data)
         batch = {
+            'id': id,
             'text': [],
             'topic': []
         }
         for topic in topics:
           batch['text'].append(text)
           batch['topic'].append(topic)