learnable-ai / main.py
ishworrsubedii's picture
Added new features and improved code formatting:
32a0eda
raw
history blame contribute delete
944 Bytes
import uvicorn
from fastapi import FastAPI
from src.apis.language_translation_api import language_translation_router
from src.apis.img_processing_api import image_processing_router
from src.apis.speech_processing_api import speech_transcription_router
import argparse
learnable_ai = FastAPI()
learnable_ai.include_router(language_translation_router, prefix='/translator', tags=["Language Translation"])
learnable_ai.include_router(image_processing_router, prefix='/image', tags=["Image Processing"])
learnable_ai.include_router(speech_transcription_router, prefix='/speech', tags=["Speech Processing"])
parser = argparse.ArgumentParser(description='LearnableAI API')
parser.add_argument('--host', type=str, default='0.0.0.0', help='Host IP address')
parser.add_argument('--port', type=int, default=7860, help='Port number')
args = parser.parse_args()
if __name__ == "__main__":
uvicorn.run(learnable_ai, host=args.host, port=args.port)