Update app.py

app.py

@@ -15,11 +15,13 @@ import argparse
 import gradio as gr
 from timeit import default_timer as timer
 import torch
+import spaces
 import numpy as np
 import pandas as pd
 import whisper
 
 
+@spaces.GPU
 whisper_model = whisper.load_model("medium").to("cuda")
 tts_model = VitsModel.from_pretrained("facebook/mms-tts-pol")
 tts_model.to("cuda")
@@ -41,7 +43,6 @@ def read_txt():
 ##### Chat z LLAMA ####
 ##### Chat z LLAMA ####
 
-
 def _load_model_tokenizer():
     model_id = 'tangger/Qwen-7B-Chat'
     tokenizer = AutoTokenizer.from_pretrained(model_id,  trust_remote_code=True)