Albert NIYONSENGA commited on
Commit
d7f762d
·
1 Parent(s): 57b0abc
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -3,9 +3,14 @@ import streamlit as st
3
  from transformers import AutoTokenizer, AutoModelForCausalLM
4
  import torch
5
  from huggingface_hub import login
6
-
7
- # Authenticate using environment variable
8
- login(os.getenv("HF_TOKEN"))
 
 
 
 
 
9
 
10
 
11
  # Sidebar
 
3
  from transformers import AutoTokenizer, AutoModelForCausalLM
4
  import torch
5
  from huggingface_hub import login
6
+
7
+ # Automatically use the environment variable
8
+ token = os.getenv("Fela")
9
+ if token:
10
+ login(token)
11
+ else:
12
+ raise ValueError("Fela environment variable is not set.")
13
+ login(os.getenv("Fela"))
14
 
15
 
16
  # Sidebar