Update app.py

app.py

@@ -10,6 +10,10 @@ model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 
 chat_history = {}
 
+@app.get("/")
+async def root():
+    return {"message": "🟢 API is live! Use `/ai?query=Hello&user_id=123`"}
+
 @app.get("/ai")
 async def chat(request: Request):
     query_params = dict(request.query_params)
@@ -24,9 +28,4 @@ async def chat(request: Request):
     response = tokenizer.decode(output_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
 
     chat_history[user_id] = [bot_input_ids, output_ids]
-    return JSONResponse({"reply": response})
-
-# ✅ Add this to launch properly on Hugging Face Spaces
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    return JSONResponse({"reply": response})