From 909f0afa692bcf8bafeec4e75a473628c002761b Mon Sep 17 00:00:00 2001 From: Alex Date: Tue, 15 Aug 2023 14:33:17 +0100 Subject: [PATCH] ruff fixes --- application/app.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/application/app.py b/application/app.py index bcf2b4665..27a99eab2 100644 --- a/application/app.py +++ b/application/app.py @@ -14,7 +14,7 @@ from celery.result import AsyncResult from flask import Flask, request, render_template, send_from_directory, jsonify, Response from langchain import FAISS -from langchain import VectorDBQA, HuggingFaceHub, Cohere, OpenAI +from langchain import VectorDBQA, Cohere, OpenAI from langchain.chains import LLMChain, ConversationalRetrievalChain from langchain.chains.conversational_retrieval.prompts import CONDENSE_QUESTION_PROMPT from langchain.chains.question_answering import load_qa_chain @@ -50,7 +50,7 @@ gpt_model = 'gpt-3.5-turbo' -if settings.SELF_HOSTED_MODEL == True: +if settings.SELF_HOSTED_MODEL: from langchain.llms import HuggingFacePipeline from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline