Update app.py

app.py

@@ -1,4 +1,3 @@
-import os
 import torch
 import torch.nn as nn
 import gradio as gr
@@ -11,10 +10,6 @@ from matplotlib.font_manager import FontProperties
 
 from model.tinyvad import TinyVAD
 
-# Configuration
-os.environ['CUDA_VISIBLE_DEVICES'] = '0'
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-
 # Font configuration
 font_path = './fonts/Times_New_Roman.ttf'
 font_prop = FontProperties(fname=font_path, size=18)
@@ -27,7 +22,7 @@ TARGET_SAMPLE_RATE = 16000
 
 # Model Initialization
 model = TinyVAD(1, 32, 64, patch_size=8, num_blocks=2, 
-                sinc_conv=SINC_CONV, ssm=SSM).to(device)
+                sinc_conv=SINC_CONV, ssm=SSM)
 checkpoint_path = './sincvad.ckpt'
 model.load_state_dict(torch.load(checkpoint_path, weights_only=True))
 model.eval()
@@ -196,7 +191,6 @@ with gr.Blocks() as demo:
         api_name="predict"
     )
 
-
 # Launch Configuration
 if __name__ == "__main__":
     demo.queue()  # Enable queue to support generators