diff --git a/a2t-whisper/main.py b/a2t-whisper/main.py index 8f7ad61..fed4134 100644 --- a/a2t-whisper/main.py +++ b/a2t-whisper/main.py @@ -4,9 +4,8 @@ from dotenv import load_dotenv from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware -from routes import router as api_router - from models import WhisperBasedModel +from routes import router as api_router load_dotenv() diff --git a/a2t-whisper/routes.py b/a2t-whisper/routes.py index 1bb0de9..10c0e70 100644 --- a/a2t-whisper/routes.py +++ b/a2t-whisper/routes.py @@ -1,9 +1,8 @@ import os from fastapi import APIRouter, UploadFile -from pydantic import BaseModel - from models import WhisperBasedModel as inference_model +from pydantic import BaseModel class AudioTranscriptionResponse(BaseModel): diff --git a/cdr-replit/main.py b/cdr-replit/main.py index c5b7d74..0a8496f 100644 --- a/cdr-replit/main.py +++ b/cdr-replit/main.py @@ -4,9 +4,8 @@ from dotenv import load_dotenv from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware -from routes import router as api_router - from models import ReplitBasedModel +from routes import router as api_router load_dotenv() diff --git a/cdr-replit/routes.py b/cdr-replit/routes.py index 2c8efac..4ed5a99 100644 --- a/cdr-replit/routes.py +++ b/cdr-replit/routes.py @@ -4,9 +4,8 @@ from typing import List, Optional, Union from fastapi import APIRouter -from pydantic import BaseModel - from models import ReplitBasedModel as model +from pydantic import BaseModel class ChatCompletionInput(BaseModel): diff --git a/cdr-t5/main.py b/cdr-t5/main.py index c5c0e92..209c2cf 100644 --- a/cdr-t5/main.py +++ b/cdr-t5/main.py @@ -4,9 +4,8 @@ from dotenv import load_dotenv from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware -from routes import router as api_router - from models import T5BasedModel +from routes import router as api_router load_dotenv() diff --git a/cdr-t5/routes.py b/cdr-t5/routes.py index 1bceeac..6615957 100644 --- a/cdr-t5/routes.py +++ b/cdr-t5/routes.py @@ -4,9 +4,8 @@ from typing import List, Optional, Union from fastapi import APIRouter -from pydantic import BaseModel - from models import T5BasedModel as model +from pydantic import BaseModel class ChatCompletionInput(BaseModel): diff --git a/cht-dolly-v2/main.py b/cht-dolly-v2/main.py index b785173..6883005 100644 --- a/cht-dolly-v2/main.py +++ b/cht-dolly-v2/main.py @@ -4,9 +4,8 @@ from dotenv import load_dotenv from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware -from routes import router as api_router - from models import DollyBasedModel +from routes import router as api_router load_dotenv() diff --git a/cht-dolly-v2/routes.py b/cht-dolly-v2/routes.py index 88251da..01b3c77 100644 --- a/cht-dolly-v2/routes.py +++ b/cht-dolly-v2/routes.py @@ -5,9 +5,8 @@ from fastapi import APIRouter, HTTPException from fastapi.responses import StreamingResponse -from pydantic import BaseModel - from models import DollyBasedModel as model +from pydantic import BaseModel class ChatCompletionInput(BaseModel): diff --git a/cht-falcon/main.py b/cht-falcon/main.py index 311286c..af75595 100644 --- a/cht-falcon/main.py +++ b/cht-falcon/main.py @@ -5,9 +5,8 @@ from dotenv import load_dotenv from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware -from routes import router as api_router - from models import FalconBasedModel +from routes import router as api_router load_dotenv() diff --git a/cht-falcon/routes.py b/cht-falcon/routes.py index fa95b47..66d187c 100644 --- a/cht-falcon/routes.py +++ b/cht-falcon/routes.py @@ -6,9 +6,8 @@ from fastapi import APIRouter, HTTPException from fastapi.responses import StreamingResponse -from pydantic import BaseModel - from models import FalconBasedModel as model +from pydantic import BaseModel class ChatCompletionInput(BaseModel): diff --git a/cht-llama-cpp/routes.py b/cht-llama-cpp/routes.py index 10a9b4c..5e47307 100644 --- a/cht-llama-cpp/routes.py +++ b/cht-llama-cpp/routes.py @@ -6,9 +6,8 @@ from fastapi import APIRouter, HTTPException from fastapi.responses import StreamingResponse -from pydantic import BaseModel - from models import LLaMACPPBasedModel as model +from pydantic import BaseModel class ChatCompletionInput(BaseModel): diff --git a/dfs-diffusers/main.py b/dfs-diffusers/main.py index 2077057..5816607 100644 --- a/dfs-diffusers/main.py +++ b/dfs-diffusers/main.py @@ -4,9 +4,8 @@ from dotenv import load_dotenv from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware -from routes import router as api_router - from models import DiffuserBasedModel +from routes import router as api_router load_dotenv() diff --git a/dfs-diffusers/routes.py b/dfs-diffusers/routes.py index 31de9bd..eff435f 100644 --- a/dfs-diffusers/routes.py +++ b/dfs-diffusers/routes.py @@ -2,9 +2,8 @@ from typing import List, Union from fastapi import APIRouter -from pydantic import BaseModel - from models import DiffuserBasedModel as model +from pydantic import BaseModel class ImageGenerationInput(BaseModel): diff --git a/ebd-all-minilm/main.py b/ebd-all-minilm/main.py index b2eefa6..a624f51 100644 --- a/ebd-all-minilm/main.py +++ b/ebd-all-minilm/main.py @@ -4,9 +4,8 @@ from dotenv import load_dotenv from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware -from routes import router as api_router - from models import SentenceTransformerBasedModel +from routes import router as api_router load_dotenv() diff --git a/ebd-all-minilm/routes.py b/ebd-all-minilm/routes.py index b6c0150..31dbc05 100644 --- a/ebd-all-minilm/routes.py +++ b/ebd-all-minilm/routes.py @@ -2,9 +2,8 @@ import tiktoken from fastapi import APIRouter -from pydantic import BaseModel - from models import SentenceTransformerBasedModel as model +from pydantic import BaseModel class EmbeddingsInput(BaseModel): diff --git a/t2a-bark/main.py b/t2a-bark/main.py index a9e1dba..8bc11b0 100644 --- a/t2a-bark/main.py +++ b/t2a-bark/main.py @@ -5,9 +5,8 @@ from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware from fastapi.staticfiles import StaticFiles -from routes import router as api_router - from models import BarkBasedModel +from routes import router as api_router load_dotenv() diff --git a/t2a-bark/routes.py b/t2a-bark/routes.py index d58e2f1..408a03a 100644 --- a/t2a-bark/routes.py +++ b/t2a-bark/routes.py @@ -2,11 +2,10 @@ from bark import SAMPLE_RATE from fastapi import APIRouter +from models import BarkBasedModel as model from pydantic import BaseModel from scipy.io.wavfile import write as write_wav -from models import BarkBasedModel as model - class AudioGenerationInput(BaseModel): prompt: str