Update app.py
Browse files
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)
|