highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-07-12 14:12:53 +02:00
Code Issues Packages Projects Releases Wiki Activity

Remove un-needed imports (#999)

Browse Source
This commit is contained in:
Chris Weaver
2024-01-25 12:10:19 -08:00
committed by GitHub
parent 92628357df
commit e94fd8b022
3 changed files with 57 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
backend/model_server/custom_models.py
View File

@ -1,5 +1,4 @@
import numpy as np
import tensorflow as tf # type:ignore
from fastapi import APIRouter
from danswer.search.search_nlp_models import get_intent_model_tokenizer
@ -13,6 +12,8 @@ router = APIRouter(prefix="/custom")
@log_function_time(print_only=True)
def classify_intent(query: str) -> list[float]:
import tensorflow as tf # type:ignore
tokenizer = get_intent_model_tokenizer()
intent_model = get_local_intent_model()
model_input = tokenizer(query, return_tensors="tf", truncation=True, padding=True)