Update main.py

main.py

@@ -3,7 +3,7 @@ from langchain.memory import ConversationBufferMemory
 from langchain.utilities import GoogleSearchAPIWrapper
 from langchain.agents import AgentType, initialize_agent, Tool
 from lang import G4F
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, Request, Path
 from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
 from ImageCreator import generate_image_prodia
@@ -118,7 +118,7 @@ details = {
 }
 
 @app.get("/imagen-details/{detail_id}")
-def image_detail(detail_id: int = Path(None, description="The ID of 1.model id and 2.sampler id")):
+def image_detail(detail_id: int = Path(None, description="The ID of 1.model id and 2.sampler id", gt=0, lt=3)):
     return details[detail_id]