gpu fixes

app.py

@@ -1,4 +1,5 @@
 # app.py
+import os
 import gradio as gr
 import torch
 import numpy as np
@@ -7,7 +8,6 @@ from transformers.models.whisper.modeling_whisper import WhisperEncoder
 import torch.nn as nn
 from safetensors.torch import load_file
 from huggingface_hub import hf_hub_download
-import os
 
 # === CUSTOM MODEL CLASSES ===
 class WhisperEncoderOnlyConfig(WhisperConfig):
@@ -49,13 +49,12 @@ AutoModel.register(WhisperEncoderOnlyConfig, WhisperEncoderOnlyForClassification
 # === LOAD MODEL ===
 MODEL_REPO = "tachiwin/language_classification_enconly_model_2"
 
-print("Loading model...")
+print("Loading model on CPU...")
 processor = WhisperProcessor.from_pretrained(MODEL_REPO)
 config = WhisperEncoderOnlyConfig.from_pretrained(MODEL_REPO)
 model = WhisperEncoderOnlyForClassification(config)
 
 # Load weights from safetensors
-from huggingface_hub import hf_hub_download
 weights_path = hf_hub_download(repo_id=MODEL_REPO, filename="model.safetensors")
 state_dict = load_file(weights_path)
 model.load_state_dict(state_dict)
@@ -63,7 +62,6 @@ model.eval()
 
 print("Model loaded successfully!")
 
-
 # === INFERENCE FUNCTION ===
 def predict_language(audio):
     if audio is None:
@@ -113,9 +111,8 @@ def predict_language(audio):
         {f"{code_idx}": code_conf}
     )
 
-
 # === UI COMPONENTS ===
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", secondary_hue="blue")) as demo:
+with gr.Blocks() as demo:
     gr.HTML(
         """
         <div style="text-align: center; padding: 30px; background: linear-gradient(135deg, #4f46e5 0%, #3b82f6 100%); color: white; border-radius: 15px; margin-bottom: 25px; box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1);">
@@ -171,4 +168,5 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", secondary_hue="blue"))
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(theme=gr.themes.Soft(primary_hue="indigo", secondary_hue="blue"))
+