Spaces:
Build error
Build error
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 |
|