Update app.py

app.py

@@ -1,28 +1,36 @@
-import gradio as gr
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import JSONResponse
 import whisper
 import torch
 import tempfile
 import os
+import uvicorn
+
+app = FastAPI()
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = whisper.load_model("large-v3", device=device)
 
-def transcribe_audio(audio_file):
-    if audio_file is None:
-        return {"error": "No audio file provided"}
+@app.post("/transcribe")
+async def transcribe_audio(file: UploadFile = File(...)):
+    if not file:
+        raise HTTPException(status_code=400, detail="No file provided")
     
     try:
-        result = model.transcribe(audio_file, fp16=False if device == "cpu" else True)
-        return {"text": result["text"]}
+        contents = await file.read()
+        
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
+            tmp_file.write(contents)
+            tmp_file_path = tmp_file.name
+        
+        result = model.transcribe(tmp_file_path, fp16=False if device == "cpu" else True)
+        
+        os.unlink(tmp_file_path)
+        
+        return JSONResponse({"text": result["text"]})
     
     except Exception as e:
-        return {"error": str(e)}
-
-app = gr.Interface(
-    fn=transcribe_audio,
-    inputs=gr.Audio(type="filepath"),
-    outputs=gr.JSON()
-)
+        raise HTTPException(status_code=500, detail=str(e))
 
 if __name__ == "__main__":
-    app.launch(share=False, server_name="0.0.0.0", server_port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)