ryanpdwyer commited on
Commit
683bb94
·
1 Parent(s): b195166

Updated to use token

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import torch
 
 
4
 
5
  # Retrieve the Hugging Face token from environment variables
6
  hf_token = os.environ.get("HF_TOKEN")
 
1
  import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import torch
4
+ import os
5
+
6
 
7
  # Retrieve the Hugging Face token from environment variables
8
  hf_token = os.environ.get("HF_TOKEN")