JustusI commited on
Commit
2f1b986
·
verified ·
1 Parent(s): bcb7f7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -1,5 +1,9 @@
1
  from openai import OpenAI
2
  import streamlit as st
 
 
 
 
3
 
4
  # import torch
5
  # from transformers import AutoModelForCausalLM, AutoTokenizer, StoppingCriteria, StoppingCriteriaList, TextIteratorStreamer
@@ -39,7 +43,7 @@ import streamlit as st
39
  # initialize the client
40
  client = OpenAI(
41
  base_url="https://api-inference.huggingface.co/v1",
42
- #api_key=os.environ.get('HUGGINGFACEHUB_API_TOKEN')#"hf_xxx" # Replace with your token
43
  )
44
 
45
 
 
1
  from openai import OpenAI
2
  import streamlit as st
3
+ import os
4
+ import sys
5
+ from dotenv import load_dotenv, dotenv_values
6
+ load_dotenv()
7
 
8
  # import torch
9
  # from transformers import AutoModelForCausalLM, AutoTokenizer, StoppingCriteria, StoppingCriteriaList, TextIteratorStreamer
 
43
  # initialize the client
44
  client = OpenAI(
45
  base_url="https://api-inference.huggingface.co/v1",
46
+ api_key=os.environ.get('HUGGINGFACEHUB_API_TOKEN')#"hf_xxx" # Replace with your token
47
  )
48
 
49