drrobot9 commited on
Commit
6b6a9ba
·
verified ·
1 Parent(s): be12601

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +2 -1
app/main.py CHANGED
@@ -55,7 +55,8 @@ async def generate_response(websocket: WebSocket, audio_np: np.ndarray):
55
  chat.end_turn()
56
 
57
  chat.new_turn("user")
58
- chat.add_audio(audio_np[np.newaxis, :], sampling_rate=SAMPLE_RATE)
 
59
  chat.end_turn()
60
 
61
  chat.new_turn("assistant")
 
55
  chat.end_turn()
56
 
57
  chat.new_turn("user")
58
+ audio_tensor = torch.from_numpy(audio_np[np.newaxis, :]).to(dtype=torch.float32)
59
+ chat.add_audio(audio_tensor, sampling_rate=SAMPLE_RATE)
60
  chat.end_turn()
61
 
62
  chat.new_turn("assistant")