ruff fixes

arc53 · Aug 15, 2023 · 909f0af · 909f0af
1 parent e593241
commit 909f0af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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