visit2sachin56 commited on
Commit
74e4e38
·
verified ·
1 Parent(s): d60ea8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,8 +2,8 @@
2
  import streamlit as st
3
  from transformers import AutoModelForCausalLM, AutoTokenizer,pipeline
4
 
 
5
  model_id = str(st.text_input("Enter model_id"))
6
- #model_id = "meta-llama/Meta-Llama-3-8B"
7
  tokenizer = AutoTokenizer.from_pretrained(model_id)
8
 
9
  model = AutoModelForCausalLM.from_pretrained(
 
2
  import streamlit as st
3
  from transformers import AutoModelForCausalLM, AutoTokenizer,pipeline
4
 
5
+ model_id = "meta-llama/Llama-2-7b-chat-hf"
6
  model_id = str(st.text_input("Enter model_id"))
 
7
  tokenizer = AutoTokenizer.from_pretrained(model_id)
8
 
9
  model = AutoModelForCausalLM.from_pretrained(