fastapi-demo-v1 / main.py
Arafath10's picture
Update main.py
47628cc
from fastapi import FastAPI, File, UploadFile
from fastapi.responses import StreamingResponse
import os
import io
from transformers import pipeline
pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
temp = open("model/t.txt","w")
temp.write("aaaaaaaaaaaaa")
temp.close()
temp = open("model/t.txt","r")
app = FastAPI()
@app.get("/")
def read_root():
output = pipe_flan("hi")
return {"message": str(output[0]["generated_text"])}
@app.post("/uploadfile/")
async def create_upload_file(file: UploadFile = File(...)):
try:
# Save the file with a specific name
file_path = "inputvoice.mp3"
with open(file_path, "wb") as f:
f.write(file.file.read())
# Read the content of the saved file
with open(file_path, "rb") as f:
file_content = f.read()
# Return the content as a streaming response
return StreamingResponse(io.BytesIO(file_content), media_type="audio/mpeg", headers={"Content-Disposition": "inline; filename=inputvoice.mp3"})
except PermissionError as e:
return {"error": f"PermissionError: {str(e)}"}