ShermanAI commited on
Commit
de0571b
·
1 Parent(s): efec88a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -3,11 +3,12 @@ import gradio as gr
3
  import openai
4
  import torch
5
  import os
6
-
7
- openai.api_type = os.environ.get("api_type_key")
8
- openai.api_base = os.environ.get("api_base_key")
9
- openai.api_version = os.environ.get("api_version_key")
10
- openai.api_key = os.environ.get("api_cpu_key")
 
11
 
12
  model_input_id = "openskyml/pigeon-llm"
13
  model_gen_id = "tiiuae/falcon-40b-instruct"
 
3
  import openai
4
  import torch
5
  import os
6
+ '''
7
+ openai.api_type = os.environ.get("OPENAI_API_KEY")
8
+ openai.api_base = os.environ.get("OPENAI_API_KEY")
9
+ openai.api_version = os.environ.get("OPENAI_API_KEY")
10
+ '''
11
+ openai.api_key = os.environ.get("OPENAI_API_KEY")
12
 
13
  model_input_id = "openskyml/pigeon-llm"
14
  model_gen_id = "tiiuae/falcon-40b-instruct"