ikenna1234 commited on
Commit
a33bad7
·
1 Parent(s): 599706f

Update space

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. app.py +2 -1
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: Tuned Llama Ai Interviewer
3
  emoji: 💬
4
  colorFrom: yellow
5
  colorTo: purple
@@ -9,4 +9,4 @@ app_file: app.py
9
  pinned: false
10
  ---
11
 
12
- An example chatbot using [Gradio](https://gradio.app), [`huggingface_hub`](https://huggingface.co/docs/huggingface_hub/v0.22.2/en/index), and the [Hugging Face Inference API](https://huggingface.co/docs/api-inference/index).
 
1
  ---
2
+ title: PPO LLAMA 3.2-1B Test Space
3
  emoji: 💬
4
  colorFrom: yellow
5
  colorTo: purple
 
9
  pinned: false
10
  ---
11
 
12
+ An example chatbot using [Gradio](https://gradio.app), [`huggingface_hub`](https://huggingface.co/docs/huggingface_hub/v0.22.2/en/index), and the [Hugging Face Inference API](https://huggingface.co/docs/api-inference/index).
app.py CHANGED
@@ -10,7 +10,8 @@ from history import get_history, update_history
10
  login(getenv("Token"))
11
 
12
  #name of model on huggingFace
13
- model="ikenna1234/llama_3.2_1b_rlhf"
 
14
  #"ikenna1234/EleutherAI_pythia_1b_rlhf"
15
 
16
  # Define generator pipeline
 
10
  login(getenv("Token"))
11
 
12
  #name of model on huggingFace
13
+ model="meta-llama/Llama-3.2-1B"
14
+ #"ikenna1234/llama_3.2_1b_rlhf"
15
  #"ikenna1234/EleutherAI_pythia_1b_rlhf"
16
 
17
  # Define generator pipeline