Update app.py

app.py

@@ -7,19 +7,20 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 
 def load_dataset():
     url = os.getenv("URL")
-    return requests.get(url).json()
+     r = requests.get(url)
+    if r.status_code != 200:
+        raise HTTPException(status_code=500, detail="Dataset fetch failed")
+    return r.json()
 
 async def auth(authorization: str | None = Header(default=None)):
-    if not authorization or authorization.replace("Bearer ", "") != HF_TOKEN:
+    if AUTH_TOKEN and (not authorization or authorization.replace("Bearer ", "") != AUTH_TOKEN):
         raise HTTPException(status_code=401, detail="Unauthorized")
 
 @app.get("/rows")
-async def get_rows(limit: int = 100, authorization: str | None = Header(default=None)):
+async def get_rows(limit: int = 3, authorization: str | None = Header(default=None)):
     await auth(authorization)
-    if not (1 <= limit <= 100):
-        raise HTTPException(status_code=400, detail="Invalid limit")
-
-    return load_dataset()[:limit]
+    data = load_dataset()
+    return data[:limit]
 
 @app.get("/row")
 async def get_row(id: str, authorization: str | None = Header(default=None)):
@@ -29,3 +30,6 @@ async def get_row(id: str, authorization: str | None = Header(default=None)):
         if r["id"] == id:
             return r
     raise HTTPException(status_code=404, detail="Not found")
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)