增加根据GPU动态加载模型

app.py

@@ -12,7 +12,10 @@ app = FastAPI()
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
-model_id = "openai/whisper-base.en"
+if torch.cuda.is_available():
+    model_id = "openai/whisper-small.en"
+else:
+    model_id = "openai/whisper-tiny.en"
 
 pipe = pipeline(
     "automatic-speech-recognition",
@@ -91,7 +94,7 @@ with gr.Blocks() as blocks:
     t = gr.Textbox(label="结果")
 
     b.click(handel, inputs=f, outputs=t)
-    
+
 blocks.queue(max_size=3)
 
 app = gr.mount_gradio_app(app, blocks, path="/")