ML-Learner / server /api /models /multi_rag_models.py
VashuTheGreat2's picture
Upload folder using huggingface_hub
0225ffc verified
Raw History Blame Contribute Delete
237 Bytes
from pydantic import BaseModel
from fastapi import File, UploadFile,File
class ChatRequest(BaseModel):
message: str
class UploadState(BaseModel):
uploaded_file:UploadFile=File(...)
class User(BaseModel):
thread_id:str