Update app.py

app.py

@@ -1,4 +1,4 @@
-
+import gradio as gr
 import spaces
 import soundfile as sf
 import torch
@@ -12,8 +12,6 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, VitsModel
 import torch
 import numpy as np
 import os
-import argparse
-import gradio as gr
 from timeit import default_timer as timer
 import torch
 import numpy as np
@@ -26,7 +24,6 @@ import whisper
 # tts_model.to("cuda")
 # print("TTS Loaded!")
 
-
 def load_whisper():
     return whisper.load_model("medium", device = 'cpu')
 
@@ -100,6 +97,8 @@ def _parse_text(text):
                 lines[i] = "<br>" + line
     text = "".join(lines)
     return text
+
+    
 @spaces.GPU
 def predict(_query, _chatbot, _task_history):
     print(f"User: {_parse_text(_query)}")
@@ -116,6 +115,7 @@ def predict(_query, _chatbot, _task_history):
     _task_history.append((_query, full_response))
     print(f"Qwen-7B-Chat: {_parse_text(full_response)}")
 
+@spaces.GPU
 def read_text(text):
     print("___Tekst do przeczytania!")
     inputs = tokenizer_tss(text, return_tensors="pt").to("cuda")
@@ -127,7 +127,7 @@ def read_text(text):
 
 def update_audio(text):
     return 'temp_file.wav'
-
+@spaces.GPU
 def translate(audio):
     print("__Wysyłam nagranie do whisper!")
     transcription = whisper_model.transcribe(audio, language="pl")