Update app.py

app.py

@@ -4,6 +4,7 @@ import torch
 import tempfile
 import asyncio
 import edge_tts
+import spaces
 from pydub import AudioSegment
 from threading import Thread
 from collections.abc import Iterator
@@ -45,7 +46,8 @@ async def text_to_speech(text: str) -> str:
 
     os.remove(wav_path)  # Delete the original WAV file
     return mp3_path  # Return the MP3 file path
-
+    
+@spaces.GPU
 def generate(
     message: str,
     chat_history: list[dict],
@@ -61,7 +63,7 @@ def generate(
 
     # Remove special tags
     if is_tts:
-        message = message.replace("edgetts@tts", "").strip()
+        message = message.replace("@tts", "").strip()
     elif is_text_only:
         message = message.replace("@text", "").strip()
 
@@ -100,9 +102,9 @@ def generate(
         loop = asyncio.new_event_loop()
         asyncio.set_event_loop(loop)
         audio_path = loop.run_until_complete(text_to_speech(final_output))
-        return audio_path  # Returning MP3 file path
+        return audio_path  
 
-    return final_output  # Returning text output
+    return final_output  # 
 
 demo = gr.ChatInterface(
     fn=generate,
@@ -117,7 +119,7 @@ demo = gr.ChatInterface(
     examples=[
         ["A train travels 60 kilometers per hour. If it travels for 5 hours, how far will it travel in total?"],
         ["@text What is AI?"],
-        ["edgetts@tts Explain Newton's third law of motion."],
+        ["@tts Explain Newton's third law of motion."],
         ["@text Rewrite the following sentence in passive voice: 'The dog chased the cat.'"],
     ],
     cache_examples=False,