Update app.py
Browse files
app.py
CHANGED
@@ -24,8 +24,8 @@ openai_api_key = os.environ.get("openai_api_key")
|
|
24 |
def load_models():
|
25 |
return {
|
26 |
"yolo": YOLO("best.pt"),
|
27 |
-
"embeddings": OpenAIEmbeddings(model="text-embedding-3-small",
|
28 |
-
"llm": ChatOpenAI(model="gpt-4-turbo", temperature=0.3,
|
29 |
}
|
30 |
|
31 |
models = load_models()
|
|
|
24 |
def load_models():
|
25 |
return {
|
26 |
"yolo": YOLO("best.pt"),
|
27 |
+
"embeddings": OpenAIEmbeddings(model="text-embedding-3-small",api_key=openai_api_key),
|
28 |
+
"llm": ChatOpenAI(model="gpt-4-turbo", temperature=0.3,api_key=openai_api_key)
|
29 |
}
|
30 |
|
31 |
models = load_models()
|