Volker Janz commited on
Commit
30068e0
·
1 Parent(s): ce2e69b

Authenticate

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -1,8 +1,12 @@
 
1
  from transformers import AutoTokenizer
2
  from adapters import AutoAdapterModel
 
3
  import gradio as gr
4
  import torch
5
 
 
 
6
 
7
  title = "Mental Health Chatbot"
8
  description = "This bot is using a fine-tuned version of meta-llama/Llama-2-7b-chat-hf"
 
1
+ from huggingface_hub import login
2
  from transformers import AutoTokenizer
3
  from adapters import AutoAdapterModel
4
+ import os
5
  import gradio as gr
6
  import torch
7
 
8
+ HF_TOKEN = os.getenv("HF_TOKEN")
9
+ login(token=HF_TOKEN)
10
 
11
  title = "Mental Health Chatbot"
12
  description = "This bot is using a fine-tuned version of meta-llama/Llama-2-7b-chat-hf"