iseesaw commited on
Commit
ebb94ea
1 Parent(s): b5c8494

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -2,7 +2,9 @@ import os
2
  from openai import OpenAI
3
  import gradio as gr
4
 
5
- client = OpenAI()
 
 
6
 
7
  def predict(message, history):
8
  history_openai_format = []
 
2
  from openai import OpenAI
3
  import gradio as gr
4
 
5
+ api_key = os.environ.get("OPENAI_API_KEY")
6
+ base_url = os.environ.get("OPENAI_BASE_URL")
7
+ client = OpenAI(api_key=api_key, base_url=base_url)
8
 
9
  def predict(message, history):
10
  history_openai_format = []