PercivalFletcher commited on
Commit
a83e37a
·
verified ·
1 Parent(s): 29617dc

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -5,7 +5,7 @@ import os
5
  import tempfile
6
  import asyncio
7
  import time
8
- from typing import List, Dict, Any
9
  from urllib.parse import urlparse, unquote
10
  from pathlib import Path
11
 
@@ -49,7 +49,7 @@ GROQ_MODEL_NAME = "llama3-70b-8192"
49
  EMBEDDING_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
50
 
51
  # Global variable to hold the initialized embedding model
52
- embed_model_instance: HuggingFaceEmbedding | None = None
53
 
54
  if GROQ_API_KEY == "gsk_...":
55
  print("WARNING: GROQ_API_KEY is not set. Please set it in your environment or main.py.")
 
5
  import tempfile
6
  import asyncio
7
  import time
8
+ from typing import List, Dict, Any, Union
9
  from urllib.parse import urlparse, unquote
10
  from pathlib import Path
11
 
 
49
  EMBEDDING_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
50
 
51
  # Global variable to hold the initialized embedding model
52
+ embed_model_instance: Union[HuggingFaceEmbedding, None] = None
53
 
54
  if GROQ_API_KEY == "gsk_...":
55
  print("WARNING: GROQ_API_KEY is not set. Please set it in your environment or main.py.")