Update app.py

app.py

@@ -1,5 +1,5 @@
 import uvicorn
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Request
 from pydantic import BaseModel
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
@@ -56,6 +56,12 @@ async def generate_text(input: TextInput):
 async def root():
     return {"message": "Qwen2.5-0.5B API is running!"}
 
+# Endpoint hiển thị API URL
+@app.get("/api_link")
+async def get_api_link(request: Request):
+    host = request.client.host
+    return {"api_url": f"http://{host}:7860"}
+
 # Chạy server khi file được gọi trực tiếp
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)