Update app.py

app.py

@@ -7,7 +7,7 @@ from transformers.models.whisper.tokenization_whisper import LANGUAGES
 from transformers.pipelines.audio_utils import ffmpeg_read
 
 model_id = "openai/whisper-large-v2"
-device = "cuda" if torch.cuda.is_available() else "CPU"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 
 LANGUANGE_MAP = {
@@ -62,11 +62,11 @@ LANGUANGE_MAP = {
 
 
 
-model_ckpt = "barto17/language-detection-fine-tuned-on-xlm-roberta-base"
-model = AutoModelForSequenceClassification.from_pretrained(model_ckpt)
-tokenizer = AutoTokenizer.from_pretrained(model_ckpt)
-
 def detect_language(sentence):
+    
+    model_ckpt = "barto17/language-detection-fine-tuned-on-xlm-roberta-base"
+    model = AutoModelForSequenceClassification.from_pretrained(model_ckpt)
+    tokenizer = AutoTokenizer.from_pretrained(model_ckpt)
     tokenized_sentence = tokenizer(sentence, return_tensors='pt')
     output = model(**tokenized_sentence)
     predictions = torch.nn.functional.softmax(output.logits, dim=-1)