Nymbo commited on
Commit
15551b4
1 Parent(s): f635187

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,11 +6,11 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
6
  import gradio as gr
7
  from threading import Thread
8
 
9
- MODEL_LIST = ["meta-llama/Llama-3.2-3B-Instruct"]
10
  HF_TOKEN = os.environ.get("HF_TOKEN", None)
11
  MODEL = os.environ.get("MODEL_ID")
12
 
13
- TITLE = "<h1><center>Llama 3.2 3B</center></h1>"
14
 
15
  PLACEHOLDER = """
16
  <center>
 
6
  import gradio as gr
7
  from threading import Thread
8
 
9
+ MODEL_LIST = ["meta-llama/Llama-3.2-1B-Instruct"]
10
  HF_TOKEN = os.environ.get("HF_TOKEN", None)
11
  MODEL = os.environ.get("MODEL_ID")
12
 
13
+ TITLE = "<h1><center>Llama 3.2 1B</center></h1>"
14
 
15
  PLACEHOLDER = """
16
  <center>