adding dir via dockerfile

Dockerfile

@@ -1,7 +1,7 @@
-FROM python:3.8.1-slim-buster
+FROM python:3.8.1-slim
 
 RUN apt-get update
-RUN apt-get install build-essential -y
+RUN apt-get install ffmpeg build-essential -y
 
 # create a new user
 # RUN useradd -m -u 1000 user
@@ -41,4 +41,5 @@ RUN pip install protobuf==3.20.* --no-cache-dir
 # COPY --chown=user . $HOME/code
 # RUN ls -la $HOME/code
 # RUN chown -r 777:777 /code
+RUN mkdir flagged
 CMD ["python3", "app.py"]

app.py

@@ -33,15 +33,18 @@ class DFSeparationApp:
 
     def predict(self, audio_file):
         # Load the audio file
-        audio_tensor = torch.tensor(audio_file[0]).to(self.device)
+        print(f"[LOG] Audio file: {audio_file}")
+        audio_tensor = torch.tensor(audio_file[1],dtype=torch.float).unsqueeze(0)
+        print(f"[LOG] Audio tensor shape: {audio_tensor.shape}")
         with torch.no_grad():
             # Make prediction
             output = self.model(audio_tensor)
-            preds = output.argmax(dim=-1)
             probs = output.softmax(dim=-1)
+            preds = probs.argmax(dim=-1)
             print(f"[LOG] Prediction: {preds.item()}")
             print(f"[LOG] Probability: {probs.max().item()}")
-        return preds.item(), probs.max().item()
+        pred_str = "Fake" if preds.item() == 1 else "Real"
+        return pred_str, probs.max().item()
     
     def run(self):
         print(f"[LOG] Running the app...")