ammoncoder123 commited on
Commit
5362559
·
verified ·
1 Parent(s): f928045

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +5 -0
chatbot.py CHANGED
@@ -1,6 +1,11 @@
1
  import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, BitsAndBytesConfig
3
  import torch
 
 
 
 
 
4
 
5
  # ================= CACHE THE MODEL =================
6
  @st.cache_resource
 
1
  import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, BitsAndBytesConfig
3
  import torch
4
+ from huggingface_hub import login
5
+ import os
6
+
7
+ # Force authentication
8
+ login(token=os.getenv("HF_TOKEN"))
9
 
10
  # ================= CACHE THE MODEL =================
11
  @st.cache_resource