NHZ commited on
Commit
9c121db
·
verified ·
1 Parent(s): 477ae2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,7 +15,7 @@ import streamlit as st
15
 
16
  # Custom wrapper for Groq API
17
  class GroqLLM(LLM):
18
- api_key: str = Field(..., env="GROQ_API_KEY")
19
  model: str = "llama-3.3-70b-versatile"
20
 
21
  @property
@@ -39,8 +39,8 @@ class GroqLLM(LLM):
39
  data = response.json()
40
  return data["choices"][0]["message"]["content"]
41
 
42
- # Initialize Groq API LLM
43
- llm = GroqLLM()
44
 
45
  # Function to extract content from a public Google Drive PDF link
46
  def extract_pdf_content(drive_url):
 
15
 
16
  # Custom wrapper for Groq API
17
  class GroqLLM(LLM):
18
+ api_key: str
19
  model: str = "llama-3.3-70b-versatile"
20
 
21
  @property
 
39
  data = response.json()
40
  return data["choices"][0]["message"]["content"]
41
 
42
+ # Initialize Groq API LLM with the provided key
43
+ llm = GroqLLM(api_key="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")
44
 
45
  # Function to extract content from a public Google Drive PDF link
46
  def extract_pdf_content(drive_url):