Update app.py

app.py

@@ -18,11 +18,6 @@ import numpy as np
 import pandas as pd
 import whisper
 
-
-# whisper_model = whisper.load_model("medium").to("cuda")
-# tts_model = VitsModel.from_pretrained("facebook/mms-tts-pol")
-# tts_model.to("cuda")
-# print("TTS Loaded!")
 DESCRIPTION = """\
 # Ai Trek - Generative AI usage
 This Space demonstrates LAIONBOT functionalities, 
@@ -166,6 +161,8 @@ def predict(audio, _chatbot, _task_history):
     audio_file = read_text(_parse_text(full_response))  # Generowanie audio
     return full_response
 
+
+@spaces.GPU
 def regenerate(_chatbot, _task_history):
     if not _task_history:
         yield _chatbot