Spaces:
Runtime error
Runtime error
Update app/main.py
Browse files- app/main.py +4 -2
app/main.py
CHANGED
@@ -272,6 +272,7 @@ import asyncio
|
|
272 |
import logging
|
273 |
import signal
|
274 |
import uvicorn
|
|
|
275 |
|
276 |
from fastapi import FastAPI, Request, HTTPException, status
|
277 |
from pydantic import BaseModel, Field
|
@@ -289,9 +290,10 @@ from contextlib import asynccontextmanager
|
|
289 |
MODEL_NAME = "FreedomIntelligence/Apollo-7B"
|
290 |
MODEL_FILE = "Apollo-7B.Q4_K_S.gguf"
|
291 |
EMBEDDING_MODEL = "Omartificial-Intelligence-Space/GATE-AraBert-v1"
|
292 |
-
QDRANT_URL = "https://12efeef2-9f10-4402-9deb-f070977ddfc8.eu-central-1-0.aws.cloud.qdrant.io:6333"
|
293 |
-
QDRANT_API_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIn0.Jb39rYQW2rSE9RdXrjdzKY6T1RF44XjdQzCvzFkjat4"
|
294 |
COLLECTION_NAME = "arabic_rag_collection"
|
|
|
|
|
|
|
295 |
|
296 |
# === LOGGING === #
|
297 |
logging.basicConfig(level=logging.DEBUG)
|
|
|
272 |
import logging
|
273 |
import signal
|
274 |
import uvicorn
|
275 |
+
import os
|
276 |
|
277 |
from fastapi import FastAPI, Request, HTTPException, status
|
278 |
from pydantic import BaseModel, Field
|
|
|
290 |
MODEL_NAME = "FreedomIntelligence/Apollo-7B"
|
291 |
MODEL_FILE = "Apollo-7B.Q4_K_S.gguf"
|
292 |
EMBEDDING_MODEL = "Omartificial-Intelligence-Space/GATE-AraBert-v1"
|
|
|
|
|
293 |
COLLECTION_NAME = "arabic_rag_collection"
|
294 |
+
QDRANT_URL = os.getenv("QDRANT_URL", "https://12efeef2-9f10-4402-9deb-f070977ddfc8.eu-central-1-0.aws.cloud.qdrant.io:6333")
|
295 |
+
QDRANT_API_KEY = os.getenv("QDRANT_API_KEY", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIn0.Jb39rYQW2rSE9RdXrjdzKY6T1RF44XjdQzCvzFkjat4")
|
296 |
+
|
297 |
|
298 |
# === LOGGING === #
|
299 |
logging.basicConfig(level=logging.DEBUG)
|