Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,6 +28,8 @@ load_dotenv()
|
|
| 28 |
|
| 29 |
base_url="https://api-inference.huggingface.co/v1"
|
| 30 |
|
|
|
|
|
|
|
| 31 |
|
| 32 |
#Create supported models
|
| 33 |
model_links ={
|
|
@@ -38,7 +40,7 @@ model_links ={
|
|
| 38 |
# "Llama-2":"meta-llama/Llama-2-7b-chat-hf"
|
| 39 |
}
|
| 40 |
|
| 41 |
-
headers = {"Authorization":
|
| 42 |
|
| 43 |
#Pull info about the model to display
|
| 44 |
model_info ={
|
|
|
|
| 28 |
|
| 29 |
base_url="https://api-inference.huggingface.co/v1"
|
| 30 |
|
| 31 |
+
API_KEY = os.environ.get('HUGGINGFACE_API')
|
| 32 |
+
|
| 33 |
|
| 34 |
#Create supported models
|
| 35 |
model_links ={
|
|
|
|
| 40 |
# "Llama-2":"meta-llama/Llama-2-7b-chat-hf"
|
| 41 |
}
|
| 42 |
|
| 43 |
+
headers = {"Authorization":"Bearer "+API_KEY}
|
| 44 |
|
| 45 |
#Pull info about the model to display
|
| 46 |
model_info ={
|