ammoncoder123 commited on
Commit
2d72ee6
·
verified ·
1 Parent(s): 2374ad2

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +5 -1
chatbot.py CHANGED
@@ -1,11 +1,15 @@
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
7
  def load_model():
8
- model_id = "ammoncoder123/IPTchatbotModel1-1.7B" # ← YOUR REAL REPO (copy-paste this exactly!)
9
 
10
  st.write("Loading tokenizer...")
11
  tokenizer = AutoTokenizer.from_pretrained(model_id)
 
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
+ # Authenticate with secret token
8
+ login(token=os.getenv("HF_TOKEN"))
9
  # ================= CACHE THE MODEL =================
10
  @st.cache_resource
11
  def load_model():
12
+ model_id = "ammoncoder123/IPTchatbotModel1-1.7B"
13
 
14
  st.write("Loading tokenizer...")
15
  tokenizer = AutoTokenizer.from_pretrained(model_id)