rca123456 commited on
Commit
6f0863d
·
verified ·
1 Parent(s): a52e725

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -13,10 +13,14 @@ import gradio as gr
13
  import os
14
  import uuid
15
 
 
 
 
 
16
  # ✅ Groq LLM Wrapper
17
  class GroqLLM(LLM):
18
  model: str = "llama3-8b-8192"
19
- api_key: str = os.environ.get("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") # Use env variable for safety
20
  temperature: float = 0.0
21
 
22
  def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str:
 
13
  import os
14
  import uuid
15
 
16
+ from dotenv import load_dotenv
17
+ load_dotenv()
18
+
19
+
20
  # ✅ Groq LLM Wrapper
21
  class GroqLLM(LLM):
22
  model: str = "llama3-8b-8192"
23
+ api_key: str = os.environ.get("GROQ_API_KEY") # Use env variable for safety
24
  temperature: float = 0.0
25
 
26
  def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str: