ExternalVendorCode/lollms-webui
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms-webui.git synced 2025-06-15 22:08:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed error

Browse Source
This commit is contained in:
Saifeddine ALOUI
2024-01-01 03:36:44 +01:00
parent 99fd4cd53b
commit 59e4e48b91
2 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
scripts/python/lollms_installer/main.py
View File

@ -36,8 +36,6 @@ ASCIIColors.red(" LoLLMS configuratoin tool")
ASCIIColors.yellow(f"Root dir : {root_path}")
app = FastAPI(debug=True)
class InstallProperties(BaseModel):
mode: str
# Serve the index.html file for all routes
@app.get("/{full_path:path}")
@ -50,13 +48,16 @@ async def serve_index(request: Request, full_path: Path):
return FileResponse(root_path/"scripts/python/lollms_installer/frontend/dist"/full_path)
return FileResponse(root_path/"scripts/python/lollms_installer/frontend/dist/index.html")
app.mount("/", StaticFiles(directory=root_path/"scripts/python/lollms_installer/frontend/dist"), name="static")
# app.mount("/", StaticFiles(directory=root_path/"scripts/python/lollms_installer/frontend/dist"), name="static")
class InstallProperties(BaseModel):
mode: str
@app.post("/start_installing")
def start_installing(data: InstallProperties):
if data=="cpu":
config.enable_gpu=False
if data.mode=="cpu":
config.enable_gpu=False
config.save_config()
else:
config.enable_gpu=True