Update app.py

app.py

@@ -25,6 +25,7 @@ def load_model(ckpt_path: str):
         wav2vec2=wav2vec2,
         tokenizer=tokenizer,
         feature_extractor=feature_extractor,
+        map_location='cpu'
     )
 
     return model