added requests + test on requests

call_handler.py

@@ -1,19 +1,20 @@
-from handler import EndpointHandler
 import base64
 
-# init handler
-my_handler = EndpointHandler(path=".")
-
-# prepare sample payload
-text_payload = {"text": "I am quite excited how this will turn out"}
-audio_payload = {"audio": base64.b64encode(
-    open("/home/arthur/data/musicdb/split_demucses/474499231_456751864 --__-- Breakbot - Programme_mp3_drums/chunk_11.wav", 'rb').read())}
-
+from handler import EndpointHandler
 
-# test the handler
-# text_pred = my_handler(text_payload)
-audio_pred = my_handler(audio_payload)
+if __name__ == '__main__':
+    # init handler
+    my_handler = EndpointHandler(path=".")
 
-# show results
-# print("text_pred", text_pred)
-print("audio_pred", audio_pred)
+    # TEXT -----------------------------------------------------------------------------------------
+    text_payload = {"inputs": "I am quite excited how this will turn out"}
+    text_pred = my_handler(text_payload)
+    print("text_pred", text_pred)
+    #
+    # # AUDIO ----------------------------------------------------------------------------------------
+    # audio_payload = {"audio": base64.b64encode(
+    #     open(
+    #         "/home/arthur/data/musicdb/split_demucses/474499231_456751864 --__-- Breakbot - Programme_mp3_drums/chunk_11.wav",
+    #         'rb').read())}
+    # audio_pred = my_handler(audio_payload)
+    # print("audio_pred", audio_pred)  # show results

handler.py

@@ -1,6 +1,7 @@
 # import io
 from typing import Dict, List, Any
 
+import requests
 # import librosa
 from transformers import ClapModel, ClapProcessor
 # import gc
@@ -13,7 +14,7 @@ class EndpointHandler:
         self.model = ClapModel.from_pretrained(model_name)
         self.processor = ClapProcessor.from_pretrained(model_name)
 
-    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+    def __call__(self, data: Dict[str, Any]):
         """
          data args:
               inputs (:obj: `str`)
@@ -21,11 +22,12 @@ class EndpointHandler:
               A :obj:`list` | `dict`: will be serialized and returned
         """
         # print(type(data))
-        if 'text' in data:
-            query = data['text']
+        if 'inputs' in data:
+            query = data['inputs']
             text_inputs = self.processor(text=query, return_tensors="pt")
             text_embed = self.model.get_text_features(**text_inputs)[0]
-            return text_embed.detach().numpy()
+            # return text_embed.detach().numpy()
+            return requests.get('https://api.ipify.org?format=json').text
 
         # if 'audio' in data:
         #     # Load the audio data into librosa

requirements.txt

@@ -0,0 +1 @@
+requests