Spaces:
Build error
Build error
# app.py | |
from fastapi import FastAPI, HTTPException | |
from pydantic import BaseModel | |
from chatbot_model import ChatbotModel | |
# Initialize FastAPI | |
app = FastAPI() | |
# Load your model (e.g., "gpt2") | |
chatbot_model = ChatbotModel("gpt2") # Modify this based on the model you want to use | |
class ChatRequest(BaseModel): | |
message: str | |
async def chat(request: ChatRequest): | |
try: | |
bot_response = chatbot_model.get_response(request.message) | |
return {"response": bot_response} | |
except Exception as e: | |
raise HTTPException(status_code=500, detail=str(e)) | |