Update app.py

app.py

@@ -3,6 +3,7 @@ import whisper
 import torch
 from transformers import pipeline
 import warnings
+import spaces
 warnings.filterwarnings("ignore")
 
 
@@ -15,11 +16,12 @@ generator = pipeline(
     "text-generation",
     model=model_name,
     device_map="auto",
-    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+    dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
 )
 
 print("Models ready.")
 
+@spaces.GPU
 def transcribe(audio_path: str) -> str:
     """Transcribe audio file to text using Whisper."""
     if audio_path is None:
@@ -27,7 +29,7 @@ def transcribe(audio_path: str) -> str:
     result = whisper_model.transcribe(audio_path)
     return result["text"].strip()
 
-
+@spaces.GPU
 def reflect(transcript: str) -> str:
     """Generate a journaling reflection using Qwen."""
     if not transcript: