0xdant commited on
Commit
e15fa4b
·
1 Parent(s): 79d82e7

Minor fixes

Browse files
Files changed (2) hide show
  1. src/app.py +1 -1
  2. src/worker_huggingface.py +3 -3
src/app.py CHANGED
@@ -2,7 +2,7 @@ import logging
2
  import os
3
  from flask import Flask, render_template, request, jsonify
4
  from flask_cors import CORS
5
- import src.worker_huggingface as worker_huggingface # Import the worker module
6
 
7
  # Initialize Flask app and CORS
8
  app = Flask(__name__)
 
2
  import os
3
  from flask import Flask, render_template, request, jsonify
4
  from flask_cors import CORS
5
+ import worker_huggingface # Import the worker module
6
 
7
  # Initialize Flask app and CORS
8
  app = Flask(__name__)
src/worker_huggingface.py CHANGED
@@ -2,10 +2,10 @@ import os
2
  import torch
3
  from langchain_core.prompts import PromptTemplate
4
  from langchain.chains import RetrievalQA
5
- from langchain.embeddings import HuggingFaceInstructEmbeddings
6
- from langchain.document_loaders import PyPDFLoader
7
  from langchain.text_splitter import RecursiveCharacterTextSplitter
8
- from langchain.vectorstores import Chroma
9
  from langchain_huggingface import HuggingFaceEndpoint
10
 
11
  # Check for GPU availability and set the appropriate device for computation.
 
2
  import torch
3
  from langchain_core.prompts import PromptTemplate
4
  from langchain.chains import RetrievalQA
5
+ from langchain_community.embeddings import HuggingFaceInstructEmbeddings
6
+ from langchain_community.document_loaders import PyPDFLoader
7
  from langchain.text_splitter import RecursiveCharacterTextSplitter
8
+ from langchain_community.vectorstores import Chroma
9
  from langchain_huggingface import HuggingFaceEndpoint
10
 
11
  # Check for GPU availability and set the appropriate device for computation.