klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-03 21:29:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Prevent Zoom on Input in Web Interface. Document Pip upgrade in Readme

Browse Source 
- Name /Reload API Controller Reload
This commit is contained in:
Debanjum Singh Solanky
2022-08-05 03:51:34 +03:00
parent 675e821d95
commit 49ef741d4b
3 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/main.py
View File

@@ -143,7 +143,7 @@ def search(q: str, n: Optional[int] = 5, t: Optional[SearchType] = None, r: Opti
@app.get('/reload')
def regenerate(t: Optional[SearchType] = None):
def reload(t: Optional[SearchType] = None):
global model
device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu")
model = initialize_search(config, regenerate=False, t=t, device=device)