Update app.py

app.py

@@ -3,21 +3,18 @@ import sys
 os.system("pip install transformers==4.27.0")
 from transformers import pipeline, WhisperModel, WhisperTokenizer, AutoModelForCTC
 os.system("pip install evaluate")
-import evaluate
-os.system("pip install evaluate[evaluator]")
+#import evaluate
+#os.system("pip install evaluate[evaluator]")
 os.system("pip install datasets")
-os.system("pip install llvmlite")
-os.system("pip install spicy==1.8.1")
+# os.system("pip install llvmlite")
+# os.system("pip install spicy==1.8.1")
 os.system("pip install soundfile")
 os.system("pip install jiwer")
 os.system("pip install datasets[audio]")
-#os.system("pip install numpy==1.21.4")
-#os.system("pip install numpy==1.22.1")
 os.system("pip install numba==0.51.2")
 from evaluate import evaluator
-from datasets import load_dataset, Audio
-from datasets import disable_caching
-from datasets import set_caching_enabled
+from datasets import load_dataset, Audio, disable_caching, set_caching_enabled
+
 set_caching_enabled(False)
 disable_caching()
 
@@ -27,7 +24,7 @@ p = pipeline("automatic-speech-recognition")
 
 huggingface_token = os.environ["huggingface_token"]
 
-whisper_miso=WhisperModel.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
+whisper_miso=WhisperModel.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token, device_map="auto")
 miso_tokenizer = WhisperTokenizer.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
 #miso_tokenizer = WhisperTokenizer.from_pretrained("openai/whisper-small", use_auth_token=huggingface_token)