deveix commited on
Commit
cb75233
1 Parent(s): 04f4397

save temp file

Browse files
Files changed (1) hide show
  1. app/main.py +2 -2
app/main.py CHANGED
@@ -295,7 +295,7 @@ async def handle_cnn(file: UploadFile = File(...)):
295
  print("got into request")
296
  print(file.content_type)
297
  # Ensure that we are handling an MP3 file
298
- if file.content_type == "audio/mpeg" or file.content_type == "audio/mp3":
299
  file_extension = ".mp3"
300
  elif file.content_type == "audio/wav":
301
  file_extension = ".wav"
@@ -310,7 +310,7 @@ async def handle_cnn(file: UploadFile = File(...)):
310
  # Save file to a temporary file if needed or process directly from memory
311
  with open(temp_filename, "wb") as f:
312
  f.write(contents)
313
-
314
  spectrograms = []
315
 
316
  clips = preprocess_audio_cnn(temp_filename)
 
295
  print("got into request")
296
  print(file.content_type)
297
  # Ensure that we are handling an MP3 file
298
+ if file.content_type in ["audio/mpeg", "audio/mp3", "application/octet-stream"]:
299
  file_extension = ".mp3"
300
  elif file.content_type == "audio/wav":
301
  file_extension = ".wav"
 
310
  # Save file to a temporary file if needed or process directly from memory
311
  with open(temp_filename, "wb") as f:
312
  f.write(contents)
313
+ print(f"File saved as {temp_filename}")
314
  spectrograms = []
315
 
316
  clips = preprocess_audio_cnn(temp_filename)