AjithBharadwaj commited on
Commit
31a1f3b
·
verified ·
1 Parent(s): dd36b8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def main():
39
  word_count = st.sidebar.slider("Number of Words", min_value=50, max_value=1000, value=200, step=50)
40
 
41
  if st.sidebar.button("Generate Blog"):
42
- model_id = "mistralai/Mistral-7B-Instruct-v0.2"
43
  tokenizer = AutoTokenizer.from_pretrained(model_id)
44
  model = AutoModelForCausalLM.from_pretrained(model_id)
45
  pipe = pipeline("text-generation", model=model, tokenizer=tokenizer,max_new_tokens=1000)
 
39
  word_count = st.sidebar.slider("Number of Words", min_value=50, max_value=1000, value=200, step=50)
40
 
41
  if st.sidebar.button("Generate Blog"):
42
+ model_id = "mistralai/Mistral-7B-Instruct-v0.1"
43
  tokenizer = AutoTokenizer.from_pretrained(model_id)
44
  model = AutoModelForCausalLM.from_pretrained(model_id)
45
  pipe = pipeline("text-generation", model=model, tokenizer=tokenizer,max_new_tokens=1000)