zliang commited on
Commit
533d3ec
·
verified ·
1 Parent(s): 4522002

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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",openai_api_key=openai_api_key),
28
- "llm": ChatOpenAI(model="gpt-4-turbo", temperature=0.3,openai_api_key=openai_api_key)
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()