Siddhant commited on
Commit
fd27c98
1 Parent(s): b9d404b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -130,8 +130,9 @@ def transcribe(stream, new_chunk):
130
  do_sample=False,
131
  )
132
  print(output)
133
- generated_text = output
134
- torch.mps.empty_cache()
 
135
 
136
  chat.append({"role": "assistant", "content": generated_text})
137
  text_str=generated_text
 
130
  do_sample=False,
131
  )
132
  print(output)
133
+ generated_text = output[0]['generated_text'][-1]["content"]
134
+ print(generated_text)
135
+ # torch.mps.empty_cache()
136
 
137
  chat.append({"role": "assistant", "content": generated_text})
138
  text_str=generated_text