Shanat commited on
Commit
2e30912
1 Parent(s): ec286c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -7,10 +7,12 @@ Created on Sat Oct 5 16:41:22 2024
7
 
8
  import gradio as gr
9
  from transformers import pipeline
 
10
 
11
 
12
  #chatbot = pipeline(model="NCSOFT/Llama-3-OffsetBias-8B")
13
- chatbot = pipeline(model="meta-llama/Llama-3.2-1B")
 
14
  #chatbot = pipeline(model="facebook/blenderbot-400M-distill")
15
 
16
  message_list = []
 
7
 
8
  import gradio as gr
9
  from transformers import pipeline
10
+ import os
11
 
12
 
13
  #chatbot = pipeline(model="NCSOFT/Llama-3-OffsetBias-8B")
14
+ token = os.getenv("HF_TOKEN")
15
+ chatbot = pipeline(model="meta-llama/Llama-3.2-1B", use_auth_token=token)
16
  #chatbot = pipeline(model="facebook/blenderbot-400M-distill")
17
 
18
  message_list = []