heikowagner commited on
Commit
578a5f5
1 Parent(s): 8acd439

Update app/run.py

Browse files
Files changed (1) hide show
  1. app/run.py +3 -0
app/run.py CHANGED
@@ -1,7 +1,10 @@
1
  # %%
2
  import load_model
3
  import cloudpickle
 
4
 
 
 
5
  # %%
6
  # llm = load_model.load_gpu_model("decapoda-research/llama-7b-hf")
7
  llm= load_model.load_openai_model()
 
1
  # %%
2
  import load_model
3
  import cloudpickle
4
+ import os
5
 
6
+ with open('.openapikey', 'r') as reader:
7
+ os.environ['OPEN_API_KEY']=reader.read()
8
  # %%
9
  # llm = load_model.load_gpu_model("decapoda-research/llama-7b-hf")
10
  llm= load_model.load_openai_model()