Update the MODEL_NAME = "llama-3.1-8b-instant"

#4
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -14,10 +14,16 @@ from groq import Groq
14
  # Configuration
15
  # -----------------------------
16
  GROQ_API_KEY = os.getenv("GROQ_API_KEY")
17
- MODEL_NAME = "llama3-8b-8192"
 
 
 
 
 
 
18
 
19
  client = Groq(api_key=GROQ_API_KEY)
20
- embedder = SentenceTransformer("all-MiniLM-L6-v2")
21
 
22
  # -----------------------------
23
  # PDF Processing
 
14
  # Configuration
15
  # -----------------------------
16
  GROQ_API_KEY = os.getenv("GROQ_API_KEY")
17
+
18
+ if not GROQ_API_KEY:
19
+ raise RuntimeError(
20
+ "GROQ_API_KEY is not set. Please add it in Hugging Face Spaces > Settings > Environment Variables."
21
+ )
22
+
23
+ MODEL_NAME = "llama-3.1-8b-instant"
24
 
25
  client = Groq(api_key=GROQ_API_KEY)
26
+
27
 
28
  # -----------------------------
29
  # PDF Processing