llama-cpp-server

Paused

matthoffner commited on Feb 11, 2024

Commit

25a22d9

verified ·

1 Parent(s): 60f34ef

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -1,6 +1,7 @@
 from llama_cpp.server.app import create_app, Settings
 from fastapi.responses import HTMLResponse
 from fastapi.middleware.cors import CORSMiddleware
 import os
 app = create_app(
@@ -18,18 +19,11 @@ app.add_middleware(
     allow_methods=["*"],
     allow_headers=["*"],
 )
-@app.get('/', response_class=HTMLResponse)
-def custom_index_route():
-    html_content = """
-    <html>
-        <body>
-            <h1>Test</h1>
-        </body>
-    </html>
-    """
-    return HTMLResponse(content=html_content)
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="https://matthoffner-llama-cpp-server.hf.space", port=7860)

 from llama_cpp.server.app import create_app, Settings
 from fastapi.responses import HTMLResponse
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import RedirectResponse
 import os
 app = create_app(
     allow_methods=["*"],
     allow_headers=["*"],
 )
+@app.get("/")
+async def redirect_root_to_docs():
+    return RedirectResponse("/docs")
 if __name__ == "__main__":
     import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)