khalednabawi11 commited on
Commit
a5c1730
·
verified ·
1 Parent(s): ca01e21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -3
app.py CHANGED
@@ -1,6 +1,12 @@
1
- import os
2
- from fastapi import FastAPI
3
- from pydantic import BaseModel
 
 
 
 
 
 
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")