DearGreen commited on
Commit
30354e1
1 Parent(s): 45604d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- generator = pipeline(model="meta-llama/Llama-2-7b-chat-hf")
 
5
  prompt = st.text_area('Your prompt: ')
6
  res = generator(prompt)
7
 
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ # generator = pipeline(model="meta-llama/Llama-2-7b-chat-hf") # gated model
5
+ generator = pipeline(model="TheBloke/CapybaraHermes-2.5-Mistral-7B-GPTQ")
6
  prompt = st.text_area('Your prompt: ')
7
  res = generator(prompt)
8