Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,12 @@
|
|
1 |
-
import
|
2 |
-
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
from langdetect import detect
|
5 |
|
6 |
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, GenerationConfig
|
@@ -9,6 +15,9 @@ from langchain.embeddings import HuggingFaceEmbeddings
|
|
9 |
from langchain.chains import RetrievalQA
|
10 |
from langchain.llms import HuggingFacePipeline
|
11 |
from qdrant_client import QdrantClient
|
|
|
|
|
|
|
12 |
|
13 |
# Get environment variables
|
14 |
QDRANT_API_KEY = os.getenv("QDRANT_API_KEY")
|
|
|
1 |
+
import torch
|
2 |
+
import asyncio
|
3 |
+
import logging
|
4 |
+
import signal
|
5 |
+
import uvicorn
|
6 |
+
import os
|
7 |
+
|
8 |
+
from fastapi import FastAPI, Request, HTTPException, status
|
9 |
+
from pydantic import BaseModel, Field
|
10 |
from langdetect import detect
|
11 |
|
12 |
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, GenerationConfig
|
|
|
15 |
from langchain.chains import RetrievalQA
|
16 |
from langchain.llms import HuggingFacePipeline
|
17 |
from qdrant_client import QdrantClient
|
18 |
+
from langchain.callbacks.base import BaseCallbackHandler
|
19 |
+
from huggingface_hub import hf_hub_download
|
20 |
+
from contextlib import asynccontextmanager
|
21 |
|
22 |
# Get environment variables
|
23 |
QDRANT_API_KEY = os.getenv("QDRANT_API_KEY")
|