freddyaboulton HF Staff commited on
Commit
63fe67d
·
verified ·
1 Parent(s): bf67dce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ import av
8
  import gradio as gr
9
  import spaces
10
  import torch
11
- from fastrtc import AdditionalOutputs, ReplyOnPause, WebRTC, WebRTCData, get_hf_turn_credentials
12
  from gradio.processing_utils import save_audio_to_cache
13
  from gradio.utils import get_upload_folder
14
  from transformers import AutoModelForImageTextToText, AutoProcessor
@@ -205,7 +205,7 @@ def generate(data: WebRTCData, history: list[dict], system_prompt: str = "", max
205
  files = []
206
  audio_file = None
207
  if data.audio is not None and data.audio[1].size > 0:
208
- audio_file = save_audio_to_cache(data.audio[1], data.audio[0], format="mp3", cache_dir=get_upload_folder())
209
  files.append(audio_file)
210
  if image is not None:
211
  files.append(image)
 
8
  import gradio as gr
9
  import spaces
10
  import torch
11
+ from fastrtc import AdditionalOutputs, ReplyOnPause, WebRTC, WebRTCData, get_hf_turn_credentials, audio_to_int16
12
  from gradio.processing_utils import save_audio_to_cache
13
  from gradio.utils import get_upload_folder
14
  from transformers import AutoModelForImageTextToText, AutoProcessor
 
205
  files = []
206
  audio_file = None
207
  if data.audio is not None and data.audio[1].size > 0:
208
+ audio_file = save_audio_to_cache(audio_to_int16(data.audio[1]), data.audio[0], format="mp3", cache_dir=get_upload_folder())
209
  files.append(audio_file)
210
  if image is not None:
211
  files.append(image)