Update app.py

app.py

@@ -364,9 +364,6 @@ class AudioUpscaler:
 @spaces.GPU(duration=300)
 def inference(audio_file, model_name, guidance_scale, ddim_steps, seed):
     audiosr = build_model(model_name=model_name)
-    
-    if torch.cuda.is_avaible():
-        torch.cuda.empty_cache() # empty cuda cache
 
     gc.collect()
 
@@ -416,8 +413,7 @@ def upscale_audio(
     Returns:
         tuple: Upscaled audio data and sample rate.
     """
-    if torch.cuda.is_avaible():
-        torch.cuda.empty_cache()
+    torch.cuda.empty_cache()
 
     gc.collect()
     upscaler = AudioUpscaler()
@@ -434,8 +430,8 @@ def upscale_audio(
         input_cutoff=input_cutoff,
         )
 
-    if torch.cuda.is_avaible():
-        torch.cuda.empty_cache()
+
+    torch.cuda.empty_cache()
 
     gc.collect()