JetzeJ commited on
Commit
6d69133
·
1 Parent(s): 79b97f8

Switch to OpenAI gpt-4o-mini

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -52,13 +52,13 @@ def build_agent():
52
  except Exception as e:
53
  return f"Error downloading file: {e}"
54
 
55
- groq_api_key = os.getenv("GROQ_API_KEY")
56
- if not groq_api_key:
57
- raise ValueError("GROQ_API_KEY environment variable not set. Add it as a Secret in your HF Space settings.")
58
 
59
  model = LiteLLMModel(
60
- model_id="groq/llama-3.3-70b-versatile",
61
- api_key=groq_api_key,
62
  temperature=0.0,
63
  )
64
 
 
52
  except Exception as e:
53
  return f"Error downloading file: {e}"
54
 
55
+ openai_api_key = os.getenv("OPENAI_API_KEY")
56
+ if not openai_api_key:
57
+ raise ValueError("OPENAI_API_KEY environment variable not set. Add it as a Secret in your HF Space settings.")
58
 
59
  model = LiteLLMModel(
60
+ model_id="openai/gpt-4o-mini",
61
+ api_key=openai_api_key,
62
  temperature=0.0,
63
  )
64