Update app.py
Browse files
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 |
-
|
|
|
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 = []
|