asimsultan commited on
Commit
453889e
·
1 Parent(s): 540f1c5

Updated docker file

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -23,6 +23,11 @@ app.add_middleware(
23
  class ChatRequest(BaseModel):
24
  message: str
25
 
 
 
 
 
 
26
  @app.post("/chat")
27
  async def chat(req: ChatRequest):
28
  try:
 
23
  class ChatRequest(BaseModel):
24
  message: str
25
 
26
+ @app.get("/")
27
+ def root():
28
+ return {"message": "Gemma Chat API is running 🚀. Use POST /chat to send messages."}
29
+
30
+
31
  @app.post("/chat")
32
  async def chat(req: ChatRequest):
33
  try: