Spaces:
Runtime error
Runtime error
Minor fixes
Browse files- src/app.py +1 -1
- 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
|
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
|
6 |
-
from
|
7 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
8 |
-
from
|
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.
|