Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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 |
|