Modifying to accept file URL from the Supabase

handler.py

@@ -5,6 +5,9 @@ import torchaudio
 import torch.nn.functional as F
 from typing import Dict, List, Any
 from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
+import requests
+import tempfile
+import os
 
 from transformers.models.wav2vec2.modeling_wav2vec2 import (
     Wav2Vec2PreTrainedModel,
@@ -145,15 +148,36 @@ class EndpointHandler():
         outputs = [{"label": self.config.id2label[i], "score": score} for i, score in enumerate(scores)]
         return outputs
 
-    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+    def download_file(self, url):
         """
-        The actual method called during inference. Expects data to have a 'path' to the audio file.
+        Downloads the file from the given URL and returns the path to the saved temporary file.
         """
-        # Get the path to the audio file from the request data
-        path = data.get("path")
+        response = requests.get(url)
+        if response.status_code == 200:
+            # Create a temporary file
+            temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.wav')
+            temp_file.write(response.content)
+            temp_file.close()
+            return temp_file.name
+        else:
+            return None
 
-        # If the path is provided, we run the prediction, else return an error message
-        if path:
-            return self.predict(path)
+    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+        """
+        The method called during inference. Expects data to have a 'url' to the audio file.
+        """
+        # Get the URL to the audio file from the request data
+        url = data.get("url")
+
+        # If the URL is provided, download the file and run the prediction
+        if url:
+            file_path = self.download_file(url)
+            if file_path:
+                output = self.predict(file_path)
+                # Optionally, delete the temporary file
+                os.remove(file_path)
+                return output
+            else:
+                return {"error": "Could not download the file from the provided URL."}
         else:
-            return {"error": "Path to the audio file is required."}
+            return {"error": "URL to the audio file is required."}